debug

spotify · Oct 2, 2023 · 4443112 · 4443112
1 parent 101d78a
commit 4443112
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/.github/workflows/tox.yml b/.github/workflows/tox.yml
@@ -13,7 +13,7 @@ jobs:
         os:
           - ubuntu-latest
           - windows-latest
-          - macos-13
+          - macos-latest
         py:
           - "3.11"
           - "3.10"

diff --git a/basic_pitch/inference.py b/basic_pitch/inference.py
@@ -153,7 +153,10 @@ 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}))  # type: ignore
+            print(f"isfinite: {np.all(np.isfinite(x))}", flush=True)
+            print(f"shape: {x.shape}", flush=True)
+            print(f"dtype: {x.dtype}", flush=True)
+            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:

diff --git a/tox.ini b/tox.ini
@@ -7,7 +7,7 @@ usedevelop = True
 [testenv]
 deps = -e .[test]
 commands =
-    pytest tests --ignore tests/test_nn.py {posargs}
+    pytest tests --ignore tests/test_nn.py -s {posargs}
 setenv =
     SOURCE = {toxinidir}/basic_pitch
     TEST_SOURCE = {toxinidir}/tests