Merge pull request #694 from analysiscenter/fix_decay_dict

SergeyTsimfer · web-flow · commit ffee5c789ef4 · 2023-05-18T18:30:10.000+03:00
Fix bug; Update version to 0.8.3
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
`@@ -26,4 +26,4 @@`
`26`	`26`	`from .utils_telegram import TelegramMessage`
`27`	`27`
`28`	`28`
`29`		`-__version__ = '0.8.2'`
	`29`	`+__version__ = '0.8.3'`