flax
[JAX] Replace uses of `isinstance(x, jax.numpy.ndarray)` with `isinstance(x, (numpy.ndarray, jax.numpy.ndarray))` where they were leading to test failures.
#1549

Merged

[JAX] Replace uses of `isinstance(x, jax.numpy.ndarray)` with `isinstance(x, (numpy.ndarray, jax.numpy.ndarray))` where they were leading to test failures. #1549

copybara-service merged 1 commit into main from test_397105832

google-cla added cla: yes

copybara-service force pushed 4 years ago

[JAX] Replace uses of `isinstance(x, jax.numpy.ndarray)` with `isinst…

bbbbdcc9

copybara-service force pushed to bbbbdcc9 4 years ago

copybara-service merged bbbbdcc9 into main 4 years ago

copybara-service deleted the test_397105832 branch 4 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone