From 2e56ee7920ab4d29bc425f1d8f94d0fd0389b823 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 1 Jul 2026 22:11:43 +0000
Subject: [PATCH] Import diffusion training schemas from models.training
 directly

The import-hoist lint flags newly re-exported names in the models/__init__.py hub as
unused (it does not treat __all__ membership as a use). Import the three diffusion
training schemas straight from models.training in routes/training.py, where they are
used in the route annotations and calls, and drop the __init__ re-export.
---
 studio/backend/models/__init__.py | 6 ------
 studio/backend/routes/training.py | 2 ++
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/studio/backend/models/__init__.py b/studio/backend/models/__init__.py
index 8ece0a569e..dbedeeca00 100644
--- a/studio/backend/models/__init__.py
+++ b/studio/backend/models/__init__.py
@@ -14,9 +14,6 @@ from .training import (
     TrainingRunDetailResponse,
     TrainingRunDeleteResponse,
     TrainingRunUpdateRequest,
-    DiffusionTrainingStartRequest,
-    DiffusionTrainingStartResponse,
-    DiffusionTrainingStatusResponse,
 )
 from .models import (
     CheckpointInfo,
@@ -75,9 +72,6 @@ from .data_recipe import (
 __all__ = [
     # Training schemas
     "TrainingStartRequest",
-    "DiffusionTrainingStartRequest",
-    "DiffusionTrainingStartResponse",
-    "DiffusionTrainingStatusResponse",
     "TrainingJobResponse",
     "TrainingStatus",
     "TrainingProgress",
diff --git a/studio/backend/routes/training.py b/studio/backend/routes/training.py
index b2ea6dfaa4..0c311a6e57 100644
--- a/studio/backend/routes/training.py
+++ b/studio/backend/routes/training.py
@@ -56,6 +56,8 @@ from models import (
     TrainingJobResponse,
     TrainingStatus,
     TrainingProgress,
+)
+from models.training import (
     DiffusionTrainingStartRequest,
     DiffusionTrainingStartResponse,
     DiffusionTrainingStatusResponse,