Skip to content

Commit 68228a7

Browse files
committed
Remove CoreML debug prints
1 parent 53f6738 commit 68228a7

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

basic_pitch/inference.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -156,9 +156,6 @@ def predict(self, x: npt.NDArray[np.float32]) -> Dict[str, npt.NDArray[np.float3
156156
if self.model_type == Model.MODEL_TYPES.TENSORFLOW:
157157
return {k: v.numpy() for k, v in cast(tf.keras.Model, self.model(x)).items()}
158158
elif self.model_type == Model.MODEL_TYPES.COREML:
159-
print(f"isfinite: {np.all(np.isfinite(x))}", flush=True)
160-
print(f"shape: {x.shape}", flush=True)
161-
print(f"dtype: {x.dtype}", flush=True)
162159
result = cast(ct.models.MLModel, self.model).predict({"input_2": x})
163160
return {
164161
"note": result["Identity_1"],

0 commit comments

Comments
 (0)