Skip to content

Commit 4de773d

Browse files
Update simple_cerebros_random_search.py
Syntax corrections...
1 parent 1dbac11 commit 4de773d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -599,7 +599,7 @@ def has_valid_metric(num):
599599
print(f"Best model name: {self.best_model_path}")
600600
return best
601601

602-
def purge_model_storage(self):
602+
def purge_model_storage(self) -> None:
603603
"""Slates all cached models.
604604
Recommended when running in a container without a mounted volume.
605605
It is recommened to use an artifiact registry to accession the best model.
@@ -608,7 +608,7 @@ def purge_model_storage(self):
608608
rmtree(model_cache_path)
609609

610610

611-
def purge_models_except_best_model(self)
611+
def purge_models_except_best_model(self) -> None:
612612
"""
613613
Recommended when running in a container without a mounted volume and building models that take considerable time to reproduce.
614614
It is recommened to use an artifiact registry to accession the best model, but this will preserve a redundant
@@ -623,7 +623,7 @@ def purge_models_except_best_model(self)
623623
for file in files_str:
624624
file_path = f"{model_cache_path}/{file}"
625625
print(f" {model_file_path}")
626-
if file_path != self.best_model_path
626+
if file_path != self.best_model_path:
627627
print(f"Removing: {file_path}")
628628
os.remove(file_path)
629629
# Temp debug code:

0 commit comments

Comments
 (0)