update docstrings

kaiko-ai · Nov 29, 2024 · 840bf4d · 840bf4d
1 parent a92f590
commit 840bf4d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/eva/core/models/wrappers/from_torchhub.py b/src/eva/core/models/wrappers/from_torchhub.py
@@ -57,7 +57,7 @@ def __init__(
 
     @override
     def load_model(self) -> None:
-        """Builds and loads the timm model as feature extractor."""
+        """Builds and loads the torch.hub model."""
         self._model: nn.Module = torch.hub.load(
             repo_or_dir=self._repo_or_dir,
             model=self._model_name,

diff --git a/tests/eva/core/models/wrappers/test_from_torchub.py b/tests/eva/core/models/wrappers/test_from_torchub.py
@@ -1,4 +1,4 @@
-"""TimmModel tests."""
+"""TorchHubModel tests."""
 
 from typing import Any, Dict, Tuple
 
@@ -66,7 +66,7 @@ def torchhub_model(
     out_indices: int | Tuple[int, ...] | None,
     model_kwargs: Dict[str, Any] | None,
 ) -> wrappers.TorchHubModel:
-    """TimmModel fixture."""
+    """TorchHubModel fixture."""
     return wrappers.TorchHubModel(
         model_name=model_name,
         repo_or_dir=repo_or_dir,