Skip to content

Commit

Permalink
update codebase
Browse files Browse the repository at this point in the history
  • Loading branch information
devlinwaluja committed Oct 8, 2024
1 parent 499f7f9 commit 70ee931
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 2 deletions.
3 changes: 2 additions & 1 deletion pipeline/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"""
Expand Down
1 change: 1 addition & 0 deletions pipeline/test_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
from pathlib import Path

import pytest

from api import APIClient
from TrafficImage import TrafficImage

Expand Down
3 changes: 2 additions & 1 deletion pipeline/test_db.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
1 change: 1 addition & 0 deletions pipeline/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit 70ee931

Please sign in to comment.