Merge pull request #694 from analysiscenter/fix_decay_dict

Fix bug; Update version to 0.8.3
analysiscenter · May 18, 2023 · ffee5c7 · ffee5c7
2 parents 60fc41c + d474a1c
commit ffee5c7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/batchflow/__init__.py b/batchflow/__init__.py
@@ -26,4 +26,4 @@
 from .utils_telegram import TelegramMessage
 
 
-__version__ = '0.8.2'
+__version__ = '0.8.3'
diff --git a/batchflow/models/torch/base.py b/batchflow/models/torch/base.py
@@ -794,7 +794,9 @@ def make_decay(self):
                 decay_dict = DECAYS_DEFAULTS.get(decay_).copy()
                 if decay == DECAYS['cos']:
                     decay_dict.update(T_max=step_params['frequency'])
-                decay_kwargs = {**decay_dict, **decay_kwargs}
+                decay_kwargs = Config({**decay_dict, **decay_kwargs})
+            else:
+                decay_kwargs = Config(decay_kwargs)
 
             # Remove unnecessary keys from kwargs
             for key in ['start_iter', 'last_iter', 'frequency']:
Original file line number	Diff line number	Diff line change
Expand Up		@@ -26,4 +26,4 @@
		from .utils_telegram import TelegramMessage


		__version__ = '0.8.2'
		__version__ = '0.8.3'