accli 0.7.5__tar.gz → 0.7.8__tar.gz

{accli-0.7.5 → accli-0.7.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: accli
-Version: 0.7.5
+Version: 0.7.8
 Summary: IIASA Accelerator Client
 Author-email: Wrufesh S <wrufesh@gmail.com>
 License: The MIT License (MIT)

{accli-0.7.5 → accli-0.7.8}/accli/AcceleratorTaskDispatcher.py

@@ -114,6 +114,7 @@ def push_folder_job(dir):
 
 class JobDispatchModel(BaseModel):
     is_holder_job: bool = True
+    build_only_task: bool = False
 
     name: str
     
@@ -143,6 +144,21 @@ class JobDispatchModel(BaseModel):
     children: List['JobDispatchModel'] = []
     callback: Optional['JobDispatchModel']
 
+    def dict(self, *args, **kwargs):
+        result = super().dict(*args, **kwargs)
+        if result['is_holder_job'] and result['execute_cluster'] == 'WKUBE':
+            if len(result['children']) > 1:
+                if not result['children'][0]['job_kwargs'].get('docker_image'):
+                    builder_task = result['children'][0].copy()
+                    
+                    builder_task['build_only_task'] = True
+
+                    builder_task['callback'] = result.copy()
+                    builder_task['name'] = f"{builder_task['callback']['name']} -- Image Builder"
+                    builder_task['callback']['name'] = f"{builder_task['callback']['name']} -- Holder"
+                    return builder_task
+        return result
+
 
 class WKubeTaskMeta(BaseModel):
     required_cores: float
@@ -234,8 +250,9 @@ class WKubeTask(GenericTask):
         wkube_task_kwargs = None
         wkube_task_meta = dict()
 
-        if (t_args or t_kwargs):
-            name = t_kwargs.pop('name')        
+        
+        name = t_kwargs.pop('name') if 'name' in t_kwargs else None 
+        if (t_args or t_kwargs):        
             WKubeTaskPydantic(*t_args, **t_kwargs)
             wkube_task_kwargs = WKubeTaskKwargs(*t_args, **t_kwargs)
             wkube_task_meta.update(WKubeTaskMeta(*t_args, **t_kwargs).dict(exclude_unset=True))

{accli-0.7.5 → accli-0.7.8}/accli/_version.py

@@ -1,3 +1,3 @@
 # Please strictly put double quote to use this info for git tag 
 # Read developer guide for git tag command with regex
-VERSION = "v0.7.5" 
+VERSION = "v0.7.8" 

{accli-0.7.5 → accli-0.7.8}/accli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: accli
-Version: 0.7.5
+Version: 0.7.8
 Summary: IIASA Accelerator Client
 Author-email: Wrufesh S <wrufesh@gmail.com>
 License: The MIT License (MIT)