Fix to issue #79. Silent progress bar in local Keras model

MolecularAI · Aug 19, 2022 · cfb7ffa · cfb7ffa
1 parent 7ec073d
commit cfb7ffa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aizynthfinder/utils/models.py b/aizynthfinder/utils/models.py
@@ -105,7 +105,7 @@ def predict(self, *args: np.ndarray, **_: np.ndarray) -> np.ndarray:
         :param args: the input vectors
         :return: the vector of the output layer
         """
-        return self.model.predict(args)
+        return self.model.predict(args, verbose=0)
 
 
 def _log_and_reraise_exceptions(method: Callable) -> Callable: