From 247dd79573fd6fe05464351ed1d098329dddeeaf Mon Sep 17 00:00:00 2001 From: Xavier Sumba Date: Sat, 8 Jan 2022 21:17:25 -0500 Subject: [PATCH] import torch to remove warning --- torchmetrics/classification/f_beta.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchmetrics/classification/f_beta.py b/torchmetrics/classification/f_beta.py index 3e63cc31337..6d5c8fe9f73 100644 --- a/torchmetrics/classification/f_beta.py +++ b/torchmetrics/classification/f_beta.py @@ -14,6 +14,7 @@ from typing import Any, Callable, Optional from warnings import warn +import torch from torch import Tensor from torchmetrics.classification.stat_scores import StatScores