Dont convert numpy array to list

spotify · Oct 2, 2023 · 101d78a · 101d78a
1 parent f85a617
commit 101d78a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/basic_pitch/inference.py b/basic_pitch/inference.py
@@ -153,7 +153,7 @@ 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:
-            return cast(ct.models.MLModel, self.model.predict({"input": x.tolist()}))  # type: ignore
+            return cast(ct.models.MLModel, self.model.predict({"input": x}))  # type: ignore
         elif self.model_type == Model.MODEL_TYPES.TFLITE:
             return self.model(input_2=x)  # type: ignore
         elif self.model_type == Model.MODEL_TYPES.ONNX: