ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-02-11 23:34:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support force_system_packages argument in k8s glue class

Browse Source
This commit is contained in:
allegroai 2023-12-26 10:12:32 +02:00
parent 432ee395e1
commit c9fc092f4e
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml_agent/glue/k8s.py
View File

@ -106,6 +106,7 @@ class K8sIntegration(Worker):
max_pods_limit=None, max_pods_limit=None,
pod_name_prefix=None, pod_name_prefix=None,
limit_pod_label=None, limit_pod_label=None,
force_system_packages=None,
**kwargs **kwargs
): ):
""" """
@ -142,7 +143,8 @@ class K8sIntegration(Worker):
self.k8s_pending_queue_name = k8s_pending_queue_name or self.K8S_PENDING_QUEUE self.k8s_pending_queue_name = k8s_pending_queue_name or self.K8S_PENDING_QUEUE
self.k8s_pending_queue_id = None self.k8s_pending_queue_id = None
self.container_bash_script = container_bash_script or self.CONTAINER_BASH_SCRIPT self.container_bash_script = container_bash_script or self.CONTAINER_BASH_SCRIPT
force_system_packages = ENV_FORCE_SYSTEM_SITE_PACKAGES.get() if force_system_packages is None:
force_system_packages = ENV_FORCE_SYSTEM_SITE_PACKAGES.get()
self._force_system_site_packages = force_system_packages if force_system_packages is not None else True self._force_system_site_packages = force_system_packages if force_system_packages is not None else True
if self._force_system_site_packages: if self._force_system_site_packages:
# Use system packages, because by we will be running inside a docker # Use system packages, because by we will be running inside a docker