pyg 2.5.2

pyg-team · Mar 20, 2024 · fbb2a50 · fbb2a50
1 parent 58c1c62
commit fbb2a50
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/pyproject.toml b/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="matthias@pyg.org"},
 ]

diff --git a/torch_geometric/__init__.py b/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',

diff --git a/torch_geometric/inspector.py b/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