pyg 2.5.2

Author: rusty1s

pyproject.toml

@@ -4,7 +4,7 @@ build-backend="flit_core.buildapi"
 
 [project]
 name="torch_geometric"
-version="2.5.1"
+version="2.5.2"
 authors=[
     {name="Matthias Fey", email="[email protected]"},
 ]

torch_geometric/__init__.py

@@ -22,7 +22,7 @@
 contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
 graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
 
-__version__ = '2.5.1'
+__version__ = '2.5.2'
 
 __all__ = [
     'EdgeIndex',

torch_geometric/inspector.py

@@ -322,13 +322,15 @@ def collect_param_data(
 
     def get_source(self, cls: Optional[Type] = None) -> str:
         r"""Returns the source code of :obj:`cls`."""
+        from torch_geometric.nn import MessagePassing
+
         cls = cls or self._cls
         if cls.__name__ in self._source_dict:
             return self._source_dict[cls.__name__]
-        try:
-            source = inspect.getsource(cls)
-        except Exception:
-            source = ''
+        if cls in {object, torch.nn.Module, MessagePassing}:
+            return ''
+        source = inspect.getsource(cls)
+
         self._source_dict[cls.__name__] = source
         return source