Fix formatting

Author: marshallweekley-eng

apps/bfd-pipeline-idr/src/idr_pipeline/model/load_progress.py

@@ -36,15 +36,24 @@ def model_type() -> ModelType:
 
     @override
     @classmethod
-    def fetch_query(
-        cls, partition: LoadPartition, start_time: datetime, source: Source, job_id: int
-    ) -> str:
+    def fetch_query(cls, partition: LoadPartition, start_time: datetime, source: Source) -> str:
         return f"""
         SELECT table_name, last_ts, last_id, batch_partition, job_start_ts,
           batch_complete_ts, job_id, max_run_ts
         FROM idr.load_progress
         WHERE table_name = %({LoadProgress.query_placeholder()})s 
         AND batch_partition = '{partition.name}'
+        AND job_id = 1
+        """
+
+    @classmethod
+    def fetch_query_by_job(cls, partition: LoadPartition, job_id: int) -> str:
+        return f"""
+        SELECT table_name, last_ts, last_id, batch_partition, job_start_ts,
+            batch_complete_ts, job_id, max_run_ts
+        FROM idr.load_progress
+        WHERE table_name = %({LoadProgress.query_placeholder()})s 
+        AND batch_partition = '{partition.name}'
         AND job_id = {job_id}
         """
 

apps/bfd-pipeline-idr/src/idr_pipeline/pipeline_utils.py

@@ -36,9 +36,7 @@
 
 def get_progress(
     load_mode: LoadMode,
-    source: Source,
     table_name: str,
-    start_time: datetime,
     partition: LoadPartition,
     job_id: int,
 ) -> LoadProgress | None:
@@ -48,7 +46,7 @@ def get_progress(
     return PostgresExtractor(
         load_mode=load_mode, cls=LoadProgress, partition=partition
     ).extract_single(
-        LoadProgress.fetch_query(partition, start_time, source, job_id),
+        LoadProgress.fetch_query_by_job(partition, job_id),
         {LoadProgress.query_placeholder(): table_name},
     )
 
@@ -77,7 +75,7 @@ def extract_and_load(
 
     while True:
         try:
-            progress = get_progress(load_mode, source, cls.table(), job_start, partition, job_id)
+            progress = get_progress(load_mode, cls.table(), partition, job_id)
 
             if progress:
                 logger.info(