Add missing return in string_to_object in InferenceConfig

AdvancedPhotonSource · May 22, 2024 · 5669a09 · 5669a09
1 parent 166a732
commit 5669a09
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/generic_trainer/configs.py b/generic_trainer/configs.py
@@ -240,6 +240,7 @@ class InferenceConfig(Config):
     def string_to_object(self, key, value):
         if key == 'model_save_dir':
             self.pretrained_model_path = os.path.join(value, 'best_model.pth')
+        return value
 
 
 @dataclasses.dataclass