From 6357fa3fd44f8b06e763599d1f65d5e9daadf989 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 25 May 2026 17:44:53 +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.14.10 → v0.15.14](https://github.com/astral-sh/ruff-pre-commit/compare/v0.14.10...v0.15.14) --- .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 0badc0b..82e507a 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -21,7 +21,7 @@ repos: - id: check-added-large-files - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.14.10 + rev: v0.15.14 # This version is automatically managed. hooks: # Check if okay now and fix From e768b35e1974bdc04f4025d35961da0200907915 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 25 May 2026 17:45:03 +0000 Subject: [PATCH 2/2] [pre-commit.ci] Auto fixes from pre-commit hooks --- src/ml_ops/link_best_model.py | 3 +-- src/ml_ops/link_model.py | 1 - src/ml_ops/promote_model.py | 3 +-- src/ml_ops/train.py | 2 +- src/ml_ops/train_cli.py | 2 +- tests/performancetests/test_model.py | 2 +- 6 files changed, 5 insertions(+), 8 deletions(-) diff --git a/src/ml_ops/link_best_model.py b/src/ml_ops/link_best_model.py index 3388e8f..7a0ef2a 100644 --- a/src/ml_ops/link_best_model.py +++ b/src/ml_ops/link_best_model.py @@ -1,9 +1,8 @@ # Takes a sweep ID, finds the best run, and links its model artifact to a model registry. import typer -from loguru import logger - import wandb +from loguru import logger app = typer.Typer() diff --git a/src/ml_ops/link_model.py b/src/ml_ops/link_model.py index 2ebd0a2..ef9a737 100644 --- a/src/ml_ops/link_model.py +++ b/src/ml_ops/link_model.py @@ -1,7 +1,6 @@ import os import typer - import wandb diff --git a/src/ml_ops/promote_model.py b/src/ml_ops/promote_model.py index 6bf6a8d..843d87c 100644 --- a/src/ml_ops/promote_model.py +++ b/src/ml_ops/promote_model.py @@ -2,9 +2,8 @@ import operator import typer -from dotenv import load_dotenv - import wandb +from dotenv import load_dotenv # Initialize Typer app app = typer.Typer() diff --git a/src/ml_ops/train.py b/src/ml_ops/train.py index d0d9f80..2b0b83c 100644 --- a/src/ml_ops/train.py +++ b/src/ml_ops/train.py @@ -5,6 +5,7 @@ import hydra import matplotlib.pyplot as plt import torch +import wandb from dotenv import load_dotenv from hydra.utils import instantiate from loguru import logger @@ -13,7 +14,6 @@ from torch.utils.data import DataLoader from torchvision.utils import make_grid -import wandb from ml_ops.data import corrupt_mnist from ml_ops.device import DEVICE from ml_ops.model import MyAwesomeModel diff --git a/src/ml_ops/train_cli.py b/src/ml_ops/train_cli.py index d72bfff..f2f95bd 100644 --- a/src/ml_ops/train_cli.py +++ b/src/ml_ops/train_cli.py @@ -8,6 +8,7 @@ import matplotlib.pyplot as plt import torch import typer +import wandb from dotenv import load_dotenv from hydra.utils import instantiate from loguru import logger @@ -16,7 +17,6 @@ from torch.utils.data import DataLoader from torchvision.utils import make_grid -import wandb from ml_ops.data import corrupt_mnist from ml_ops.device import DEVICE from ml_ops.model import MyAwesomeModel diff --git a/tests/performancetests/test_model.py b/tests/performancetests/test_model.py index ec2f20e..c497cad 100644 --- a/tests/performancetests/test_model.py +++ b/tests/performancetests/test_model.py @@ -11,8 +11,8 @@ import time import torch - import wandb + from ml_ops.model import MyAwesomeModel