Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 11 additions & 11 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -19,16 +19,16 @@ readme = "README.md"
requires-python = ">=3.12"
dependencies = [
"dvc[ssh]>=3.65.0",
"fastapi==0.115.6",
"fastapi==0.135.2",
"gdown>=5.2.0",
"hydra-core>=1.3.2",
"omegaconf>=2.3.0",
"matplotlib>=3.10.8",
"pathspec<1.0.0",
"pathspec<2.0.0",
"scikit-learn>=1.8.0",
"torch==2.9.1",
"typer==0.15.1",
"uvicorn[standard]==0.34.0",
"torch==2.11.0",
"typer==0.24.1",
"uvicorn[standard]==0.42.0",
"loguru>=0.7.3",
"wandb>=0.23.1",
"pytorch-lightning>=2.6.0",
Expand All @@ -50,13 +50,13 @@ dependencies = [
# uv sync --all-groups
[dependency-groups]
dev = [
"coverage==7.6.9",
"invoke==2.2.0",
"mkdocs-material==9.4.6",
"coverage==7.13.5",
"invoke==2.2.1",
"mkdocs-material==9.7.6",
"mkdocs==1.6.1",
"pre-commit==4.1.0",
"pytest==8.3.4",
"mkdocstrings-python==1.12.2",
"pre-commit==4.5.1",
"pytest==9.0.2",
"mkdocstrings-python==2.0.3",
"mypy>=1.19.1", # static type checker
"ruff>=0.14.10",
"ty>=0.0.9",
Expand Down
3 changes: 1 addition & 2 deletions src/ml_ops/link_best_model.py
Original file line number Diff line number Diff line change
@@ -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()

Expand Down
1 change: 0 additions & 1 deletion src/ml_ops/link_model.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import os

import typer

import wandb


Expand Down
3 changes: 1 addition & 2 deletions src/ml_ops/promote_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down
2 changes: 1 addition & 1 deletion src/ml_ops/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down
2 changes: 1 addition & 1 deletion src/ml_ops/train_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down
2 changes: 1 addition & 1 deletion tests/performancetests/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@
import time

import torch

import wandb

from ml_ops.model import MyAwesomeModel


Expand Down
970 changes: 580 additions & 390 deletions uv.lock

Large diffs are not rendered by default.

Loading