From cbb5c873f0bed671b622d453b13f71402e7a8c44 Mon Sep 17 00:00:00 2001 From: seilmast Date: Thu, 13 Feb 2025 13:32:33 +0100 Subject: [PATCH] ruff i think --- CollaborativeCoding/dataloaders/download.py | 2 +- CollaborativeCoding/load_data.py | 2 +- CollaborativeCoding/load_model.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/CollaborativeCoding/dataloaders/download.py b/CollaborativeCoding/dataloaders/download.py index d99eff2..59b6345 100644 --- a/CollaborativeCoding/dataloaders/download.py +++ b/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) diff --git a/CollaborativeCoding/load_data.py b/CollaborativeCoding/load_data.py index a3bfe42..c3a8951 100644 --- a/CollaborativeCoding/load_data.py +++ b/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) diff --git a/CollaborativeCoding/load_model.py b/CollaborativeCoding/load_model.py index 11e53c0..a76e5dd 100644 --- a/CollaborativeCoding/load_model.py +++ b/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(