Skip to content

Commit

Permalink
add other template params
Browse files Browse the repository at this point in the history
  • Loading branch information
safoinme committed Nov 14, 2023
1 parent 6073878 commit 59320d0
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions tests/test_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,11 @@ def generate_and_run_project(
model: str = "distilbert-base-uncased",
zenml_server_url: str = "",
accelerator: str = "cpu",
deploy_locally: bool = True,
deploy_to_huggingface: bool = False,
deploy_to_skypilot: bool = False,
cloud_of_choice: str = "gcp",

):
"""Generate and run the starter project with different options."""

Expand All @@ -57,6 +62,10 @@ def generate_and_run_project(
"sample_rate": sample_rate,
"model": model,
"accelerator": accelerator,
"deploy_locally": deploy_locally,
"deploy_to_huggingface": deploy_to_huggingface,
"deploy_to_skypilot": deploy_to_skypilot,
"cloud_of_choice": cloud_of_choice,
}
if open_source_license:
answers["email"] = "[email protected]"
Expand Down

0 comments on commit 59320d0

Please sign in to comment.