fix utilities.py: fixing an invalid flax import caused by the latest …

…flax release (google/flax#4177)
ivy-llc · Oct 18, 2024 · 256df76 · 256df76
1 parent 5ce47fc
commit 256df76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/stateful/utilities.py b/ivy/stateful/utilities.py
@@ -24,7 +24,7 @@ def _is_submodule(obj, kw):
             "keras.src.layers.layer.Layer",
         ),
         "flax": (
-            "flax.nnx.nnx.module.Module",
+            "flax.nnx.module.Module",
             "transformers.modeling_flax_utils.FlaxPreTrainedModel",
         ),
     }[kw]