Update function task runtimes in the workflow engine

tapis-project · Nov 7, 2024 · e1afb07 · e1afb07
1 parent 7bfc188
commit e1afb07
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/api/specs/WorkflowsAPI.yaml b/src/api/specs/WorkflowsAPI.yaml
@@ -4656,6 +4656,7 @@ components:
         - "pytorch/pytorch:latest"
         - "huggingface/transformers-pytorch-gpu:latest"
         - "huggingface/transformers-pytorch-gpu:4.29.2"
+        - "ghcr.io/tobiashi26/pygeoflood-container:main"
 
     EnumTaskIOType:
       type: string

diff --git a/src/engine/src/owe_python_sdk/constants.py b/src/engine/src/owe_python_sdk/constants.py
@@ -31,6 +31,9 @@
         "pytorch/pytorch:latest",
         "huggingface/transformers-pytorch-gpu:latest",
         "huggingface/transformers-pytorch-gpu:4.29.2"
+
+        # TACC custom runtime # TODO refactor into a plugin
+        "ghcr.io/tobiashi26/pygeoflood-container:main"
     ]
 }