Replace deprecated warn with warning

tomaarsen · Sep 29, 2023 · faa8fa1 · faa8fa1
1 parent 93415bd
commit faa8fa1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/span_marker/modeling.py b/span_marker/modeling.py
@@ -399,7 +399,7 @@ def predict(
         from span_marker.trainer import Trainer
 
         if torch.cuda.is_available() and self.device == torch.device("cpu"):
-            logger.warn(
+            logger.warning(
                 "SpanMarker model predictions are being computed on the CPU while CUDA is available."
                 " Moving the model to CUDA using `model.cuda()` before performing predictions is heavily"
                 " recommended to significantly boost prediction speeds.",