Update lock

NeilGirdhar · Jan 18, 2025 · 907b31d · 907b31d
1 parent 4052b1f
commit 907b31d
Show file tree

Hide file tree

Showing 2 changed files with 108 additions and 111 deletions.
diff --git a/tests/test_flax_display.py b/tests/test_flax_display.py
@@ -6,14 +6,13 @@
 from rich.console import Console
 
 from tjax import print_generic
-from tjax.dataclasses import field
+from tjax.dataclasses import DataClassModule, field
 
 from .test_display import verify
 
 try:
     from flax import nnx
 
-    from tjax.dataclasses import DataClassModule
 except ImportError:
     pytest.skip("Skipping NNX display tests", allow_module_level=True)