diff --git a/punchpipe/controlsegment/db.py b/punchpipe/controlsegment/db.py index 453d7cc..e8b3bc8 100644 --- a/punchpipe/controlsegment/db.py +++ b/punchpipe/controlsegment/db.py @@ -1,11 +1,7 @@ import os -from typing import Optional -from prefect.blocks.core import Block -from pydantic import SecretStr from sqlalchemy import TEXT, Column, DateTime, Integer, String from sqlalchemy.orm import declarative_base -from prefect_sqlalchemy.credentials import DatabaseCredentials Base = declarative_base() diff --git a/punchpipe/controlsegment/util.py b/punchpipe/controlsegment/util.py index 489b28a..2108d00 100644 --- a/punchpipe/controlsegment/util.py +++ b/punchpipe/controlsegment/util.py @@ -3,7 +3,6 @@ import yaml from prefect import task from punchbowl.data import PUNCHData -from sqlalchemy import create_engine from sqlalchemy.orm import Session from yaml.loader import FullLoader from prefect_sqlalchemy.credentials import DatabaseCredentials diff --git a/punchpipe/monitor/monitor.py b/punchpipe/monitor/monitor.py index 506a6cd..5026300 100644 --- a/punchpipe/monitor/monitor.py +++ b/punchpipe/monitor/monitor.py @@ -7,12 +7,12 @@ import pandas as pd import plotly.express as px from punchbowl.data import PUNCHData -from sqlalchemy import and_, create_engine, or_ +from sqlalchemy import and_, or_ from sqlalchemy.orm import Session from sqlalchemy.orm.exc import MultipleResultsFound, NoResultFound from prefect_sqlalchemy.credentials import DatabaseCredentials -from punchpipe.controlsegment.db import File, FileRelationship, Flow, MySQLCredentials +from punchpipe.controlsegment.db import File, FileRelationship, Flow def _create_overall_blocks(start_date, end_date):