Merge pull request #31 from hubverse-org/znk/nodate

prevent parquet files with date columns from borking the build process
hubverse-org · Nov 22, 2024 · 09faa07 · 09faa07
2 parents 3e1696a + f9e51fe
commit 09faa07
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/hub_predtimechart/app/generate_json_files.py b/src/hub_predtimechart/app/generate_json_files.py
@@ -111,7 +111,7 @@ def generate_forecast_json_file(hub_config, model_id_to_df, output_dir, target,
         file_name = json_file_name(target, task_ids_tuple, reference_date)
         json_file_path = output_dir / file_name
         with open(json_file_path, 'w') as fp:
-            json.dump(forecast_data, fp, indent=4)
+            json.dump(forecast_data, fp, indent=4, default=str)
             return json_file_path
 
     return None

diff --git a/tests/hubs/example-complex-forecast-hub/model-output/PSI-DICE/2022-12-17-PSI-DICE.parquet b/tests/hubs/example-complex-forecast-hub/model-output/PSI-DICE/2022-12-17-PSI-DICE.parquet