Skip to content

Commit b925fde

Browse files
Fix Git CI error.
1 parent b9da609 commit b925fde

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

apps/bfd-model-idr/load_splitter.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,13 @@
6262
type=click.Path(exists=True),
6363
default="./batched_out",
6464
)
65-
def main(clm_batches_size: int, out: Path):
65+
66+
def main_cli(clm_batches_size: int, out: Path):
67+
main(clm_batches_size=clm_batches_size, out=out)
68+
69+
def main(clm_batches_size: int, out: Path = Path("./batched_out")):
70+
"""Generate synthetic claims data."""
71+
out = Path(out) if isinstance(out, str) else out
6672
files_to_copy: dict[str, list[RowAdapter]] = {
6773
BENE_HSTRY: [],
6874
BENE_MBI_ID: [],
@@ -203,4 +209,4 @@ def main(clm_batches_size: int, out: Path):
203209

204210

205211
if __name__ == "__main__":
206-
main(clm_batches_size=5)
212+
main_cli()

0 commit comments

Comments
 (0)