diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 4936062..230ed9c 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -38,12 +38,14 @@ jobs: strategy: matrix: stack-name: [local] - os: [windows-latest, ubuntu-latest, macos-latest] + os: [windows-latest, ubuntu-latest, macos-13] python-version: ["3.8", "3.9", "3.10", "3.11"] env: ZENML_DEBUG: true ZENML_ANALYTICS_OPT_IN: false ZENML_LOGGING_VERBOSITY: INFO + PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0 #https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/9133 + PYTORCH_NO_CUDA_MEMORY_CACHING: 1 steps: - name: Check out repository code uses: actions/checkout@v3 diff --git a/template/README.md b/template/README.md index d03002a..e0c567d 100644 --- a/template/README.md +++ b/template/README.md @@ -163,7 +163,7 @@ to indicate where you can add your code: ## 📜 Project Structure -The project loosely follows [the recommended ZenML project structure](https://docs.zenml.io/user-guide/starter-guide/follow-best-practices): +The project loosely follows [the recommended ZenML project structure](https://docs.zenml.io/how-to/setting-up-a-project-repository/best-practices): ``` . diff --git a/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} b/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} index a5ef9dd..5ba624b 100644 --- a/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} +++ b/template/steps/promotion/{% if metric_compare_promotion %}promote_metric_compare_promoter.py{% endif %} @@ -34,7 +34,7 @@ def promote_metric_compare_promoter( In this example, the step can be configured to use different input data. See the documentation for more information: - https://docs.zenml.io/user-guide/advanced-guide/configure-steps-pipelines + https://docs.zenml.io/how-to/build-pipelines/use-pipeline-step-parameters Args: latest_metrics: Recently trained model metrics results. @@ -69,4 +69,4 @@ def promote_metric_compare_promoter( logger.info( f"Promoted current model version to {pipeline_extra['target_env']} environment" ) - ### YOUR CODE ENDS HERE ### \ No newline at end of file + ### YOUR CODE ENDS HERE ### diff --git a/template/steps/tokenizer_loader/tokenizer_loader.py b/template/steps/tokenizer_loader/tokenizer_loader.py index 9f43e30..e1d4952 100644 --- a/template/steps/tokenizer_loader/tokenizer_loader.py +++ b/template/steps/tokenizer_loader/tokenizer_loader.py @@ -27,7 +27,7 @@ def tokenizer_loader( For more information on how to configure steps in a pipeline, refer to the following documentation: - https://docs.zenml.io/user-guide/advanced-guide/configure-steps-pipelines + https://docs.zenml.io/how-to/build-pipelines/use-pipeline-step-parameters Args: lower_case: A boolean value indicating whether to convert the input text to diff --git a/tests/test_template.py b/tests/test_template.py index cfdd192..24648a1 100644 --- a/tests/test_template.py +++ b/tests/test_template.py @@ -28,7 +28,6 @@ TEMPLATE_DIRECTORY = str(pathlib.Path.joinpath(pathlib.Path(__file__).parent.parent)) - def generate_and_run_project( tmp_path_factory: pytest.TempPathFactory, open_source_license: Optional[str] = "apache",