Skip to content

Commit

Permalink
Merge pull request #22 from zenml-io/misc/uv-as-default-docker-builder
Browse files Browse the repository at this point in the history
UV for Docker by default
  • Loading branch information
avishniakov authored Jul 19, 2024
2 parents 37dcb95 + bfbc824 commit b807ffd
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 1 deletion.
1 change: 1 addition & 0 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ jobs:
run-tests:
runs-on: ${{ matrix.os }}
strategy:
fail-fast: false
matrix:
stack-name: [local]
os: [windows-latest, ubuntu-latest, macos-13]
Expand Down
4 changes: 4 additions & 0 deletions template/config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,11 @@ settings:
parent_image: 'huggingface/transformers-pytorch-gpu'
build_options:
platform: 'linux/amd64'
python_package_installer_args:
system: null
{%- endif %}
python_package_installer: uv
install_stack_requirements: False
required_integrations:
{%- if cloud_of_choice == 'aws' %}
- aws
Expand Down
9 changes: 8 additions & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@


import os
import sys
import shutil
from typing import Generator

Expand Down Expand Up @@ -99,4 +100,10 @@ def clean_zenml_client(

# remove all traces, and change working directory back to base path
os.chdir(orig_cwd)
shutil.rmtree(str(tmp_path))
if sys.platform == "win32":
try:
shutil.rmtree(str(tmp_path))
except:
pass
else:
shutil.rmtree(str(tmp_path))

0 comments on commit b807ffd

Please sign in to comment.