From f1fea48ab2f08e4ad97c7ea82971e7de1ad94161 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 22 Sep 2025 19:16:25 +0000 Subject: [PATCH 1/2] [pre-commit.ci] pre-commit autoupdate MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit updates: - [github.com/astral-sh/ruff-pre-commit: v0.4.4 → v0.13.1](https://github.com/astral-sh/ruff-pre-commit/compare/v0.4.4...v0.13.1) --- .pre-commit-config.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 45409c7..ea55e71 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -2,7 +2,7 @@ ci: skip: [mypy, pytest] repos: - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.4.4 + rev: v0.13.1 hooks: - id: ruff args: [--fix] From d3ee4d1ca93c742eee74d55db23857ab3636cefc Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 22 Sep 2025 19:16:34 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- ai_toolkit/datasets/__init__.py | 2 +- ai_toolkit/metrics/__init__.py | 2 +- ai_toolkit/metrics/accuracy.py | 2 +- ai_toolkit/metrics/f1_score.py | 2 +- ai_toolkit/metrics/loss.py | 2 +- ai_toolkit/models/__init__.py | 6 +++--- ai_toolkit/test.py | 2 +- 7 files changed, 9 insertions(+), 9 deletions(-) diff --git a/ai_toolkit/datasets/__init__.py b/ai_toolkit/datasets/__init__.py index 832ef8d..3d96957 100644 --- a/ai_toolkit/datasets/__init__.py +++ b/ai_toolkit/datasets/__init__.py @@ -12,7 +12,7 @@ def get_dataset_initializer(dataset_name: str) -> DatasetLoader: """Retrieves class initializer from its string name.""" if not hasattr(sys.modules[__name__], dataset_name): raise RuntimeError(f"Dataset class {dataset_name} not found in datasets/") - return cast(DatasetLoader, getattr(sys.modules[__name__], dataset_name)()) + return cast("DatasetLoader", getattr(sys.modules[__name__], dataset_name)()) __all__ = ( diff --git a/ai_toolkit/metrics/__init__.py b/ai_toolkit/metrics/__init__.py index e57ff8b..3917864 100644 --- a/ai_toolkit/metrics/__init__.py +++ b/ai_toolkit/metrics/__init__.py @@ -15,7 +15,7 @@ def get_metric_initializer(metric_name: str) -> type[Metric]: """Retrieves class initializer from its string name.""" if not hasattr(sys.modules[__name__], metric_name): raise RuntimeError(f"Metric {metric_name} not found in metrics folder.") - return cast(type[Metric], getattr(sys.modules[__name__], metric_name)) + return cast("type[Metric]", getattr(sys.modules[__name__], metric_name)) __all__ = ( diff --git a/ai_toolkit/metrics/accuracy.py b/ai_toolkit/metrics/accuracy.py index 678d506..7003593 100644 --- a/ai_toolkit/metrics/accuracy.py +++ b/ai_toolkit/metrics/accuracy.py @@ -7,7 +7,7 @@ class Accuracy(Metric): def __repr__(self) -> str: - return f"{self.name}: {100. * self.value:.2f}%" + return f"{self.name}: {100.0 * self.value:.2f}%" @staticmethod def calculate_accuracy(output: torch.Tensor, target: torch.Tensor) -> float: diff --git a/ai_toolkit/metrics/f1_score.py b/ai_toolkit/metrics/f1_score.py index fd6d0ba..dbe24ca 100644 --- a/ai_toolkit/metrics/f1_score.py +++ b/ai_toolkit/metrics/f1_score.py @@ -30,7 +30,7 @@ def calculate_f1_score( f1 = 2 * (precision * recall) / (precision + recall + eps) f1 = f1.clamp(min=eps, max=1 - eps) f1_score = 1 - f1.mean() - return cast(float, f1_score) + return cast("float", f1_score) def update(self, val_dict: SimpleNamespace) -> float: y_pred, y_true = val_dict.output, val_dict.target diff --git a/ai_toolkit/metrics/loss.py b/ai_toolkit/metrics/loss.py index fb3d36d..db5a5e4 100644 --- a/ai_toolkit/metrics/loss.py +++ b/ai_toolkit/metrics/loss.py @@ -10,4 +10,4 @@ def update(self, val_dict: SimpleNamespace) -> float: self.epoch_avg += loss * val_dict.batch_size self.running_avg += loss * val_dict.batch_size self.num_examples += val_dict.batch_size - return cast(float, loss) + return cast("float", loss) diff --git a/ai_toolkit/models/__init__.py b/ai_toolkit/models/__init__.py index 775da11..5d1f41c 100644 --- a/ai_toolkit/models/__init__.py +++ b/ai_toolkit/models/__init__.py @@ -19,14 +19,14 @@ def get_model_initializer(model_name: str) -> type[nn.Module]: """Retrieves class initializer from its string name.""" if not hasattr(sys.modules[__name__], model_name): raise RuntimeError(f"Model class {model_name} not found in models/") - return cast(type[nn.Module], getattr(sys.modules[__name__], model_name)) + return cast("type[nn.Module]", getattr(sys.modules[__name__], model_name)) __all__ = ( "BasicCNN", - "DenseNet", "BasicLSTM", - "MaskRCNN", "BasicRNN", + "DenseNet", + "MaskRCNN", "get_model_initializer", ) diff --git a/ai_toolkit/test.py b/ai_toolkit/test.py index 51cee8f..a2a1199 100644 --- a/ai_toolkit/test.py +++ b/ai_toolkit/test.py @@ -46,7 +46,7 @@ def test_model( test_loss /= test_len print( f"\nTest set: Average loss: {test_loss:.4f},", - f"Accuracy: {correct}/{test_len} ({100. * correct / test_len:.2f}%)\n", + f"Accuracy: {correct}/{test_len} ({100.0 * correct / test_len:.2f}%)\n", )