Fix torch deprecated warning

Toni-SM · Jun 21, 2024 · 9f265af · 9f265af
1 parent 36d4a57
commit 9f265af
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/skrl/resources/schedulers/torch/kl_adaptive.py b/skrl/resources/schedulers/torch/kl_adaptive.py
@@ -1,5 +1,7 @@
 from typing import Optional, Union
 
+from packaging import version
+
 import torch
 from torch.optim.lr_scheduler import _LRScheduler
 
@@ -50,6 +52,8 @@ def __init__(self,
         :param verbose: Verbose mode (default: ``False``)
         :type verbose: bool, optional
         """
+        if version.parse(torch.__version__) >= version.parse("2.2"):
+            verbose = "deprecated"
         super().__init__(optimizer, last_epoch, verbose)
 
         self.kl_threshold = kl_threshold