diff --git a/libsys_airflow/plugins/data_exports/full_dump_marc.py b/libsys_airflow/plugins/data_exports/full_dump_marc.py index d827879f..92403248 100644 --- a/libsys_airflow/plugins/data_exports/full_dump_marc.py +++ b/libsys_airflow/plugins/data_exports/full_dump_marc.py @@ -39,7 +39,9 @@ def fetch_number_of_records(**kwargs) -> int: conn_id="postgres_folio", database=kwargs.get("database", "okapi"), sql=query, - ).execute(context) + ).execute( + context + ) # type: ignore count = result[0][0] logger.info(f"Record count: {count}") diff --git a/libsys_airflow/plugins/data_exports/instance_ids.py b/libsys_airflow/plugins/data_exports/instance_ids.py index 2c2e6b1f..fdf10b43 100644 --- a/libsys_airflow/plugins/data_exports/instance_ids.py +++ b/libsys_airflow/plugins/data_exports/instance_ids.py @@ -51,7 +51,9 @@ def fetch_record_ids(**kwargs) -> dict: "from_date": from_date, "to_date": to_date, }, - ).execute(context) + ).execute( + context + ) # type: ignore ) results[kind] = list(np.unique(results[kind]))