From 70ee93153e86355212d44102b9611b9963e2cede Mon Sep 17 00:00:00 2001 From: devlinwaluja Date: Tue, 8 Oct 2024 17:19:21 +0800 Subject: [PATCH] update codebase --- pipeline/db.py | 3 ++- pipeline/test_api.py | 1 + pipeline/test_db.py | 3 ++- pipeline/test_model.py | 1 + 4 files changed, 6 insertions(+), 2 deletions(-) diff --git a/pipeline/db.py b/pipeline/db.py index 7220c9a..7a02373 100644 --- a/pipeline/db.py +++ b/pipeline/db.py @@ -9,9 +9,10 @@ import firebase_admin from firebase_admin import firestore from google.cloud.firestore import DocumentReference -from model import to_json_dict from pydantic import BaseModel +from model import to_json_dict + class DatabaseClient: """Firestore Database (DB) client""" diff --git a/pipeline/test_api.py b/pipeline/test_api.py index 619470a..b670807 100644 --- a/pipeline/test_api.py +++ b/pipeline/test_api.py @@ -8,6 +8,7 @@ from pathlib import Path import pytest + from api import APIClient from TrafficImage import TrafficImage diff --git a/pipeline/test_db.py b/pipeline/test_db.py index 929f87a..361704d 100644 --- a/pipeline/test_db.py +++ b/pipeline/test_db.py @@ -14,9 +14,10 @@ from uuid import uuid4 import pytest +from pydantic import BaseModel + from db import DatabaseClient from model import to_json_dict -from pydantic import BaseModel class Model(BaseModel): diff --git a/pipeline/test_model.py b/pipeline/test_model.py index edab23f..4c505b6 100644 --- a/pipeline/test_model.py +++ b/pipeline/test_model.py @@ -10,6 +10,7 @@ from pathlib import Path from jsonschema import validate + from model import Camera, Congestion, Location, Rating, to_json_dict CONGESTION_SCHEMA = Path(__file__).parent.parent / "schema" / "congestion.schema.json"