Update types to prevent pytype errors with an upcoming change to JAX.

magenta / mt3

MT3: Multi-Task Multitrack Music Transcription

Apache License 2.0

1.41k stars 185 forks source link

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Update types to prevent pytype errors with an upcoming change to JAX.