Revert changes

bin/src/learner/raytune_learner.py

@@ -16,7 +16,6 @@ def __init__(self, config_path: str, model_class: nn.Module, data_path: str, exp
         """
         Initialize the TuneWrapper with the paths to the config, model, and data.
         """
-        self.best_config = None
         self.config = YamlRayConfigLoader(config_path).get_config()
         self.config["model"] = model_class
         self.config["experiment"] = experiment_object
@@ -51,13 +50,6 @@ def tune(self) -> None:
         """
         return self.tuner.fit() 
 
-    def store_best_config(self, path: str) -> None:
-        """
-        Store the best config in a file.
-        """
-        with open(path, "w") as f:
-            f.write(str(self.best_config))  
-
 class TuneModel(Trainable):
 
     def setup(self, config: dict) -> None: