diff --git a/bin/src/learner/raytune_learner.py b/bin/src/learner/raytune_learner.py index 2dd9945..0eb9ffd 100644 --- a/bin/src/learner/raytune_learner.py +++ b/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: