update

pyg-team · Feb 8, 2023 · 081fd14 · 081fd14
1 parent d6577d4
commit 081fd14
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/docs/source/modules/nn.rst b/docs/source/modules/nn.rst
@@ -6,8 +6,6 @@ torch_geometric.nn
 
 .. autoclass:: torch_geometric.nn.sequential.Sequential
 
-.. autoclass:: torch_geometric.nn.model_hub.PyGModelHubMixin
-
 .. currentmodule:: torch_geometric.nn.dense
 
    {% for name in torch_geometric.nn.dense.lin_classes %}

diff --git a/torch_geometric/nn/model_hub.py b/torch_geometric/nn/model_hub.py
@@ -112,7 +112,7 @@ def save_pretrained(self, save_directory: Union[str, Path],
                 If not provided will default to the name of
                 :obj:`save_directory` in your namespace. (default: :obj:`None`)
             **kwargs: Additional keyword arguments passed to
-                :obj:`huggingface_hub.ModelHubMixin.save_pretrained`.
+                :meth:`huggingface_hub.ModelHubMixin.save_pretrained`.
         """
         config = self.model_config
         # due to way huggingface hub handles the loading/saving of models,
@@ -186,7 +186,7 @@ def from_pretrained(
         cache_dir: Optional[str] = None,
         local_files_only: bool = False,
         **model_kwargs,
-    ):
+    ) -> Any:
         r"""Downloads and instantiates a model from the HuggingFace hub.
 
         Args: