def deploy_csv_api_output(api_output: pydantic.BaseModel, output_path: pathlib.Path, columns: List[str]) -> None: if not hasattr(api_output, "__root__"): raise AssertionError("Missing root data") data = _model_to_dict(api_output.__dict__) rows = dataset_deployer.remove_root_wrapper(data) dataset_deployer.write_nested_csv(rows, output_path, header=columns)
def deploy_csv_api_output( api_output: pydantic.BaseModel, output_path: pathlib.Path, keys_to_skip: Optional[List[str]] = None, ) -> None: if not hasattr(api_output, "__root__"): raise AssertionError("Missing root data") rows = dataset_deployer.remove_root_wrapper(api_output.dict()) dataset_deployer.write_nested_csv(rows, output_path, keys_to_skip=keys_to_skip)
def deploy_csv_api_output( intervention: Intervention, api_output: pydantic.BaseModel, output_dir: pathlib.Path, filename_override=None, ): if not hasattr(api_output, "__root__"): raise AssertionError("Missing root data") if not output_dir.exists(): output_dir.mkdir(parents=True, exist_ok=True) filename = filename_override or (api_output.output_key(intervention) + ".csv") output_path = output_dir / filename rows = dataset_deployer.remove_root_wrapper(api_output.dict()) dataset_deployer.write_nested_csv(rows, output_path)
def test_remove_root(): value = AggregateRegionSummaryWithTimeseries(__root__=[]).dict() assert value == {"__root__": []} result = dataset_deployer.remove_root_wrapper(value) assert result == []
def test_remove_root(): value = CovidActNowBulkTimeseries(__root__=[]).dict() assert value == {"__root__": []} result = dataset_deployer.remove_root_wrapper(value) assert result == []