ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-04-05 21:15:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

rename to sync_extra_args and set UV_CACHE_DIR

Browse Source
This commit is contained in:
Mads Oestergaard 2024-11-04 10:05:56 +01:00
parent 3d06b410c4
commit 16a6701d2a
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clearml_agent/commands/worker.py
View File

@ -3453,8 +3453,6 @@ class Worker(ServiceCommandSection):
self.uv.initialize(cwd=lockfile_path)
api = self.uv.get_api(lockfile_path)
if api.enabled:
# TODO: what should this be? This controls where uv installs its venv
os.environ["UV_PROJECT_ENVIRONMENT"] = None
print('UV Enabled: Ignoring requested python packages, using repository uv lock file!')
api.install()
return api

6
clearml_agent/helper/package/uv_api.py
View File

@ -73,8 +73,12 @@ class UvConfig:
kwargs["env"]["PATH"] = path
if self.session and self.session.config and args and args[0] == "sync":
# Set the cache dir to venvs dir
if (cache_dir := self.session.config.get("agent.venvs_dir", None)) is not None:
os.environ["UV_CACHE_DIR"] = cache_dir
extra_args = self.session.config.get(
"agent.package_manager.uv_install_extra_args", None
"agent.package_manager.uv_sync_extra_args", None
)
if extra_args:
args = args + tuple(extra_args)