fix (#1542)

Deci-AI · Oct 17, 2023 · 39cae16 · 39cae16
1 parent db39011
commit 39cae16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/super_gradients/common/deprecate.py b/src/super_gradients/common/deprecate.py
@@ -112,7 +112,7 @@ def override(self, **entries):
 
     def decorator(func):
         def wrapper(*args, **training_params):
-            if deprecated_tparam_name in training_params:
+            if training_params.get(deprecated_tparam_name):
                 import super_gradients
 
                 is_still_supported = parse_version(super_gradients.__version__) < parse_version(removed_from)