Merge pull request #161 from matangover/remove-debug-prints

Remove CoreML debug prints
spotify · Jan 17, 2025 · f423902 · f423902
2 parents 53f6738 + 68228a7
commit f423902
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/basic_pitch/inference.py b/basic_pitch/inference.py
@@ -156,9 +156,6 @@ def predict(self, x: npt.NDArray[np.float32]) -> Dict[str, npt.NDArray[np.float3
         if self.model_type == Model.MODEL_TYPES.TENSORFLOW:
             return {k: v.numpy() for k, v in cast(tf.keras.Model, self.model(x)).items()}
         elif self.model_type == Model.MODEL_TYPES.COREML:
-            print(f"isfinite: {np.all(np.isfinite(x))}", flush=True)
-            print(f"shape: {x.shape}", flush=True)
-            print(f"dtype: {x.dtype}", flush=True)
             result = cast(ct.models.MLModel, self.model).predict({"input_2": x})
             return {
                 "note": result["Identity_1"],