Skip to content

Commit

Permalink
more formatting changes
Browse files Browse the repository at this point in the history
  • Loading branch information
David Rubinstein committed Sep 29, 2023
1 parent b1d20b2 commit 6b0060f
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion basic_pitch/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,9 @@ def predict(self, x: npt.NDArray[np.float32]) -> Dict[str, npt.NDArray[np.float3
elif self.model_type == Model.MODEL_TYPES.TFLITE:
return cast(tflite.Interpreter, self.model)(x) # type: ignore
elif self.model_type == Model.MODEL_TYPES.ONNX:
return cast(ort.InferenceSession, self.model).run(["note", "onset", "contour"], {"input": x}) # type: ignore
return cast(ort.InferenceSession, self.model).run( # type: ignore
["note", "onset", "contour"], {"input": x}
)


def window_audio_file(
Expand Down

0 comments on commit 6b0060f

Please sign in to comment.