diff --git a/src/cli/commands.py b/src/cli/commands.py index 0b9fc39..4186cff 100644 --- a/src/cli/commands.py +++ b/src/cli/commands.py @@ -73,14 +73,15 @@ def _get_brands_for_url(url: str) -> List[str]: def create_job_file_for_project( - project_id: int, - project_name: str, + project_id: int, + project_name: str, session, tier1_branded_ratio: Optional[float] = None, tier1_branded_text: Optional[str] = None, tier1_branded_plus_ratio: Optional[float] = None, brand_names: Optional[List[str]] = None, - random_deployment_targets: Optional[int] = None + random_deployment_targets: Optional[int] = None, + tier1_count: Optional[int] = None, ) -> Optional[str]: """ Create a job JSON file for a newly created project. @@ -1198,14 +1199,15 @@ def ingest_cora(file_path: str, name: str, money_site_url: Optional[str], custom tier1_branded_plus_ratio = None job_file = create_job_file_for_project( - project.id, - project.name, + project.id, + project.name, session, tier1_branded_ratio=tier1_branded_ratio, tier1_branded_text=tier1_branded_text, tier1_branded_plus_ratio=tier1_branded_plus_ratio, brand_names=brand_names, - random_deployment_targets=random_deployment_targets + random_deployment_targets=random_deployment_targets, + tier1_count=tier1_count, ) if job_file: click.echo(f"Job file created: {job_file}")