Update simple_cerebros_random_search.py

Inconstancy in package imports
david-thrower · Jun 12, 2024 · 36b6663 · 36b6663
1 parent cd6d6bd
commit 36b6663
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py
@@ -572,7 +572,9 @@ def run_random_search(self):
 
     def get_best_model(self):
         best_model_path = self.best_model_path
-        best_model = clone_model(tf.keras.models.load_model(best_model_path))
+        best_model =\
+            tensorflow.keras.models.clone_model(
+                tf.keras.models.load_model(best_model_path))
         # best_model = tf.keras.models.load_model(self.best_model_path)
         return best_model