File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
cerebros/simplecerebrosrandomsearch Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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:
You can’t perform that action at this time.
0 commit comments