diff --git a/pyproject.toml b/pyproject.toml index b1202a2009ef..91f199db901f 100644 --- 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 index f26808ab14fc..dcac312b4877 100644 --- 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 index e65b4d0495b3..9d24e4c70b34 100644 --- 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