Skip to content

Commit

Permalink
Fix coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
wesselb committed Sep 15, 2023
1 parent 3f82352 commit d76d28a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions lab/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,10 +82,10 @@ def _module_attr(module, attr):
_ag_tensor = ModuleType("autograd.tracer", "Box")

# Define JAX module types.
if sys.version_info.minor <= 7:
if sys.version_info.minor <= 7: # pragma: specific no cover 3.8 3.9 3.10 3.11
# `jax` 0.4 deprecated Python 3.7 support. Rely on older JAX versions.
_jax_tensor = ModuleType("jax.interpreters.xla", "DeviceArray")
else:
else: # pragma: specific no cover 3.7
_jax_tensor = ModuleType("jaxlib.xla_extension", "ArrayImpl")
_jax_tracer = ModuleType("jax.core", "Tracer")
_jax_dtype = ModuleType("jax._src.numpy.lax_numpy", "_ScalarMeta")
Expand Down

0 comments on commit d76d28a

Please sign in to comment.