Support force_system_packages argument in k8s glue class

clearml · Dec 26, 2023 · c9fc092 · c9fc092
1 parent 432ee39
commit c9fc092
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/clearml_agent/glue/k8s.py b/clearml_agent/glue/k8s.py
@@ -106,6 +106,7 @@ def __init__(
             max_pods_limit=None,
             pod_name_prefix=None,
             limit_pod_label=None,
+            force_system_packages=None,
             **kwargs
     ):
         """
@@ -142,7 +143,8 @@ def __init__(
         self.k8s_pending_queue_name = k8s_pending_queue_name or self.K8S_PENDING_QUEUE
         self.k8s_pending_queue_id = None
         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
         if self._force_system_site_packages:
             # Use system packages, because by we will be running inside a docker