import torch to remove warning

Lightning-AI · Jan 9, 2022 · 247dd79 · 247dd79
1 parent 42cafb3
commit 247dd79
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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