Skip to content

Commit

Permalink
move .readthedocs.yml to docs/
Browse files Browse the repository at this point in the history
  • Loading branch information
Borda committed Sep 16, 2024
1 parent ead5cbb commit 4125a8e
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 1 deletion.
9 changes: 9 additions & 0 deletions SandBox/issue-2656.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
from torchmetrics import MultitaskWrapper, F1Score
import torch
wrapper = MultitaskWrapper({"F1": F1Score(num_classes=2, average='macro', task="multiclass")})
wrapper2 = wrapper.clone(postfix="train")
preds = {"F1": torch.ones((5, 2))}
tgt = {"F1": torch.tensor([0, 1, 0, 1, 0], dtype=torch.long)}

#wrapper(preds, tgt) # This one works
wrapper2(preds, tgt)
2 changes: 1 addition & 1 deletion .readthedocs.yml → docs/.readthedocs.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

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

Expand Down

0 comments on commit 4125a8e

Please sign in to comment.