NetKet 3.20.1__tar.gz → 3.20.2__tar.gz

{netket-3.20.1 → netket-3.20.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: NetKet
-Version: 3.20.1
+Version: 3.20.2
 Summary: Netket : Machine Learning toolbox for many-body quantum systems.
 Project-URL: homepage, https://www.netket.org
 Project-URL: documentation, https://netket.readthedocs.io/en/latest/#

{netket-3.20.1 → netket-3.20.2}/netket/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '3.20.1'
-__version_tuple__ = version_tuple = (3, 20, 1)
+__version__ = version = '3.20.2'
+__version_tuple__ = version_tuple = (3, 20, 2)
 
 __commit_id__ = commit_id = None

{netket-3.20.1 → netket-3.20.2}/netket/utils/model_frameworks/nnx.py

@@ -80,12 +80,18 @@ class NNXWrapper:
         return nnx_module
 
     def __getattr__(self, name):
-        if hasattr(_get_graphdef_type(self.graphdef), name):
-            return partial(self.apply, method=name)
-        raise AttributeError(
-            f"{_get_graphdef_type(self.static_module)} (wrapped into a '{type(self).__name__}')"
-            f"has no attribute '{name}'"
-        )
+        # Use try-except instead of hasattr to avoid infinite recursion
+        # when checking for attributes that don't exist
+        graphdef_type = _get_graphdef_type(object.__getattribute__(self, "graphdef"))
+        try:
+            # Check if the attribute exists on the graphdef type's class
+            getattr(graphdef_type, name)
+            return partial(object.__getattribute__(self, "apply"), method=name)
+        except AttributeError:
+            raise AttributeError(
+                f"{graphdef_type} (wrapped into a '{type(self).__name__}')"
+                f" has no attribute '{name}'"
+            )
 
     def __hash__(self):
         return hash(("NNXWrapper", self.graphdef))