Skip to content

Commit 4125a8e

Browse files
committed
move .readthedocs.yml to docs/
1 parent ead5cbb commit 4125a8e

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

SandBox/issue-2656.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from torchmetrics import MultitaskWrapper, F1Score
2+
import torch
3+
wrapper = MultitaskWrapper({"F1": F1Score(num_classes=2, average='macro', task="multiclass")})
4+
wrapper2 = wrapper.clone(postfix="train")
5+
preds = {"F1": torch.ones((5, 2))}
6+
tgt = {"F1": torch.tensor([0, 1, 0, 1, 0], dtype=torch.long)}
7+
8+
#wrapper(preds, tgt) # This one works
9+
wrapper2(preds, tgt)

.readthedocs.yml renamed to docs/.readthedocs.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
# .readthedocs.yml
15+
# .readthedocs.yaml
1616
# Read the Docs configuration file
1717
# See https://docs.readthedocs.io/en/stable/config-file/v2.html for details
1818

0 commit comments

Comments
 (0)