Skip to content

Commit 90efae1

Browse files
committed
Add FastAPI app with Docker, CI/CD, and linter
1 parent 5066ffa commit 90efae1

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

test/test_api.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import os
22
import sys
33
from unittest.mock import MagicMock, patch
4+
from fastapi.testclient import TestClient
5+
from api.main import app
46

57
# Add src to sys.path for imports
68
sys.path.insert(
@@ -13,11 +15,6 @@
1315

1416
patcher = patch('mlflow.sklearn.load_model', return_value=mock_model)
1517
patcher.start()
16-
17-
from fastapi.testclient import TestClient
18-
from api.main import app
19-
20-
2118
client = TestClient(app)
2219

2320

0 commit comments

Comments
 (0)