WIP

tests/web/test_dataset_routes.py

@@ -35,6 +35,8 @@ def _make_dataset(**overrides) -> MagicMock:
         name="test-dataset",
         description="Test dataset",
         status="ready",
+        training_status=None,
+        active_training_task_id=None,
         train_ratio=0.8,
         val_ratio=0.1,
         seed=42,
@@ -183,6 +185,8 @@ class TestListDatasetsRoute:
 
         mock_db = MagicMock()
         mock_db.get_datasets.return_value = ([_make_dataset()], 1)
+        # Mock the active training tasks lookup to return empty dict
+        mock_db.get_active_training_tasks_for_datasets.return_value = {}
 
         result = asyncio.run(fn(admin_token=TEST_TOKEN, db=mock_db, status=None, limit=20, offset=0))