diff --git a/template/README.md b/template/README.md index 9a529ac..49f68e2 100644 --- a/template/README.md +++ b/template/README.md @@ -194,9 +194,9 @@ The project loosely follows [the recommended ZenML project structure](https://do │ │ ├── promote_latest.py # promote latest step │ │ ├── promote_metric_compare_promoter.py # metric compare promoter step │ │ └── promote_get_metrics.py # get metric step -│ ├── registrer # `zenml.steps.registrer` implementations +│ ├── register # `zenml.steps.register` implementations │ │ ├── __init__.py -│ │ └── model_log_registrer.py # model log registrer step +│ │ └── model_log_register.py # model log register step │ ├── tokenization # `zenml.steps.tokenization` implementations │ │ ├── __init__.py │ │ └── tokenization.py # tokenization step diff --git a/template/run.py b/template/run.py index bc27d3b..4484d0c 100644 --- a/template/run.py +++ b/template/run.py @@ -88,7 +88,7 @@ "--training-pipeline", is_flag=True, default=True, - help="Whether to run the pipeline that traines the model to {{target_environment}}.", + help="Whether to run the pipeline that train the model to {{target_environment}}.", ) @click.option( "--promoting-pipeline", diff --git a/template/steps/__init__.py b/template/steps/__init__.py index 62b6cb0..a02f657 100644 --- a/template/steps/__init__.py +++ b/template/steps/__init__.py @@ -13,7 +13,7 @@ promote_current, {%- endif %} ) -from .registrer import register_model +from .register import register_model from .tokenizer_loader import ( tokenizer_loader, ) diff --git a/template/steps/deploying/save_model.py b/template/steps/deploying/save_model.py index 8b257e0..95938c2 100644 --- a/template/steps/deploying/save_model.py +++ b/template/steps/deploying/save_model.py @@ -28,7 +28,7 @@ def save_model_to_deploy(): f" Loading latest version of the model for stage {pipeline_extra['target_env']}..." ) # Get latest saved model version in target environment - latest_version = get_step_context().model_version._get_model_version() + latest_version = get_step_context().model_version # Load model and tokenizer from Model Control Plane model = latest_version.load_artifact(name="model") diff --git a/template/steps/deploying/{% if deploy_locally %}local_deployment.py{% endif %} b/template/steps/deploying/{% if deploy_locally %}local_deployment.py{% endif %} index 9fb332d..19aa97c 100644 --- a/template/steps/deploying/{% if deploy_locally %}local_deployment.py{% endif %} +++ b/template/steps/deploying/{% if deploy_locally %}local_deployment.py{% endif %} @@ -58,7 +58,7 @@ def deploy_locally( return process.pid - lables = ",".join(labels) + labels = ",".join(labels) # Construct the path to the app.py file zenml_repo_root = Client().root if not zenml_repo_root: @@ -69,7 +69,7 @@ def deploy_locally( ) app_path = str(os.path.join(zenml_repo_root, "gradio", "app.py")) command = ["python", app_path, "--tokenizer_name_or_path", tokenizer_name_or_path, "--model_name_or_path", model_name_or_path, - "--labels", lables, "--title", title, "--description", description, + "--labels", labels, "--title", title, "--description", description, "--interpretation", interpretation, "--examples", example] logger.info(f"Command: {command}") # Call the function to launch the script diff --git a/template/steps/registrer/__init__.py b/template/steps/register/__init__.py similarity index 100% rename from template/steps/registrer/__init__.py rename to template/steps/register/__init__.py diff --git a/template/steps/registrer/model_log_register.py b/template/steps/register/model_log_register.py similarity index 100% rename from template/steps/registrer/model_log_register.py rename to template/steps/register/model_log_register.py