Skip to content

Commit

Permalink
Shorten the inference pipeline name
Browse files Browse the repository at this point in the history
  • Loading branch information
stefannica committed Nov 25, 2024
1 parent f915c1f commit 7a54214
Show file tree
Hide file tree
Showing 7 changed files with 9 additions and 9 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@ model:
```

<p align="center">
<img height=900 src="assets/06_batch_inference.png">
<img height=900 src="assets/06_inference.png">
</p>

### [Continuous Deployment] Batch Inference: ETL Steps
Expand Down
2 changes: 1 addition & 1 deletion copier.yml
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ email:
product_name:
type: str
help: The technical name of the data product you are building
default: supply_chain_forecasting
default: supply_chain_fcast
target_environment:
type: str
help: "The target environment for your project"
Expand Down
2 changes: 1 addition & 1 deletion template/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ The project loosely follows [the recommended ZenML project structure](https://do
│ ├── inference_config.yaml # the configuration of the batch inference pipeline
│ └── train_config.yaml # the configuration of the training pipeline
├── pipelines # `zenml.pipeline` implementations
│ ├── batch_inference.py # [CD] Batch Inference pipeline
│ ├── inference.py # [CD] Batch Inference pipeline
│ ├── deployment.py # [CD] Deployment pipeline
│ └── training.py # [CT] Training Pipeline
├── steps # logically grouped `zenml.steps` implementations
Expand Down
2 changes: 1 addition & 1 deletion template/pipelines/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# {% include 'template/license_header' %}


from .batch_inference import {{product_name}}_batch_inference
from .inference import {{product_name}}_inference
from .training import {{product_name}}_training
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@


@pipeline(on_failure=notify_on_failure, tags=[DATA_CLASSIFICATION])
def {{product_name}}_batch_inference():
def {{product_name}}_inference():
"""
Model batch inference pipeline.
Expand Down
6 changes: 3 additions & 3 deletions template/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from zenml.client import Client
from zenml.logger import get_logger

from pipelines import {{product_name}}_batch_inference, {{product_name}}_training
from pipelines import {{product_name}}_inference, {{product_name}}_training

logger = get_logger(__name__)

Expand Down Expand Up @@ -177,8 +177,8 @@ def main(
)
pipeline_args[
"run_name"
] = f"{{product_name}}_batch_inference_run_{dt.now().strftime('%Y_%m_%d_%H_%M_%S')}"
{{product_name}}_batch_inference.with_options(**pipeline_args)(**run_args_inference)
] = f"{{product_name}}_inference_run_{dt.now().strftime('%Y_%m_%d_%H_%M_%S')}"
{{product_name}}_inference.with_options(**pipeline_args)(**run_args_inference)



Expand Down
2 changes: 1 addition & 1 deletion tests/test_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ def generate_and_run_project(
) from e

# check the pipeline run is successful
for pipeline_suffix in ["_training", "_batch_inference"]:
for pipeline_suffix in ["_training", "_inference"]:
pipeline = Client().get_pipeline(product_name + pipeline_suffix)
assert pipeline
runs = pipeline.runs
Expand Down

0 comments on commit 7a54214

Please sign in to comment.