From 36148cfbe71ea9554c4c417f6a92e5250d2cf548 Mon Sep 17 00:00:00 2001
From: Kevin Donahue <nonnontrivial@gmail.com>
Date: Fri, 3 May 2024 16:18:44 -0400
Subject: [PATCH] fix imports

---
 api/api/tests/test_pollution_api.py  | 2 +-
 api/api/tests/test_prediction_api.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/api/tests/test_pollution_api.py b/api/api/tests/test_pollution_api.py
index 605bc31..178f050 100644
--- a/api/api/tests/test_pollution_api.py
+++ b/api/api/tests/test_pollution_api.py
@@ -1,7 +1,7 @@
 import pytest
 from fastapi.testclient import TestClient
 
-from api import app
+from api.main import app
 
 client = TestClient(app)
 API_PREFIX = "/api/v1"
diff --git a/api/api/tests/test_prediction_api.py b/api/api/tests/test_prediction_api.py
index f7a9ef5..359e103 100644
--- a/api/api/tests/test_prediction_api.py
+++ b/api/api/tests/test_prediction_api.py
@@ -1,7 +1,7 @@
 import pytest
 from fastapi.testclient import TestClient
 
-from api import app
+from api.main import app
 
 client = TestClient(app)
 API_PREFIX = "/api/v1"