Fix Output return value from imported_baseline - see HEA-784

Author: rhunwicks

pipelines/assets/base.py

@@ -314,7 +314,7 @@ def get_bss_dataframe(
     end_strings: Optional[list[str]] = None,
     header_rows: list[int] = [3, 4, 5],  # List of row indexes that contain the Wealth Group and other headers
     num_summary_cols: Optional[int] = None,
-) -> pd.DataFrame:
+) -> Output[pd.DataFrame]:
     """
     Retrieve a worksheet from a BSS and return it as a DataFrame.
 

pipelines/assets/fixtures.py

@@ -30,7 +30,7 @@
 
 def validate_instances(
     context: AssetExecutionContext, config: BSSMetadataConfig, instances: dict[str, list[dict]], partition_key: str
-) -> tuple[dict[str, list[dict]], dict]:
+) -> Output[dict[str, list[dict]]]:
     """
     Validate the instances for a set of related models, prior to loading them as a fixture.
 
@@ -322,7 +322,7 @@ def get_fixture_from_instances(instance_dict: dict[str, list[dict]]) -> tuple[li
     return Output(fixture, metadata=metadata)
 
 
-def import_fixture(fixture: list[dict]) -> dict:
+def import_fixture(fixture: list[dict]) -> Output[None]:
     """
     Import a Django fixture and return a metadata dictionary.
     """
@@ -436,9 +436,4 @@ def imported_baseline(
     """
     Imported Django fixture for a BSS, added to the Django database.
     """
-    metadata = import_fixture(consolidated_fixture)
-
-    return Output(
-        None,
-        metadata=metadata,
-    )
+    return import_fixture(consolidated_fixture)