Remove CoreML debug prints

spotify · Jan 17, 2025 · 68228a7 · 68228a7
1 parent 53f6738
commit 68228a7
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"],