ruff i think

Author: Seilmast

CollaborativeCoding/dataloaders/download.py

@@ -33,7 +33,7 @@ class Downloader:
     Examples
     --------
     >>> from pathlib import Path
-    >>> from utils import Downloader
+    >>> from CollaborativeCoding import Downloader
     >>> dir = Path('tmp')
     >>> dir.mkdir(exist_ok=True)
     >>> train, test = Downloader().usps(dir)

CollaborativeCoding/load_data.py

@@ -39,7 +39,7 @@ def load_data(dataset: str, *args, **kwargs) -> tuple:
 
     Examples
     --------
-    >>> from utils import setup_data
+    >>> from CollaborativeCoding import setup_data
     >>> train, val, test = setup_data("usps_0-6", data_path="data", train=True, download=True)
     >>> len(train), len(val), len(test)
     (4914, 546, 1782)

CollaborativeCoding/load_model.py

@@ -28,7 +28,7 @@ def load_model(modelname: str, *args, **kwargs) -> nn.Module:
 
     Examples
     --------
-    >>> from utils import load_model
+    >>> from CollaborativeCoding import load_model
     >>> model = load_model("magnusmodel", num_classes=10)
     >>> model
     MagnusModel(