fix recursive __new__ bug (#4196)

pyg-team · Mar 4, 2022 · e18a897 · e18a897
1 parent 2dfff9a
commit e18a897
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch_geometric/data/storage.py b/torch_geometric/data/storage.py
@@ -43,6 +43,9 @@ def __len__(self) -> int:
         return len(self._mapping)
 
     def __getattr__(self, key: str) -> Any:
+        if key == '_mapping':
+            self._mapping = {}
+            return self._mapping
         try:
             return self[key]
         except KeyError: