@@ -143,7 +143,12 @@ def build_ec2_type_label(key):
143143 max_active_tasks = 30 ,
144144 default_args = dag_default_args ,
145145 params = {
146- "repository" : Param ("https://github.com/unity-sds/unity-example-application" , type = "string" , title = "Repository" , description = "Repository to build from" ),
146+ "repository" : Param (
147+ "https://github.com/unity-sds/unity-example-application" ,
148+ type = "string" ,
149+ title = "Repository" ,
150+ description = "Repository to build from" ,
151+ ),
147152 "log_level" : Param (
148153 DEFAULT_LOG_LEVEL ,
149154 type = "integer" ,
@@ -167,12 +172,15 @@ def build_ec2_type_label(key):
167172)
168173
169174app_gen_env_vars = [
170- k8s .V1EnvVar (name = "DOCKERHUB_USERNAME" , value = "{{ ti.xcom_pull(task_ids='Setup', key='dockerhub_username') }}" ),
171- k8s .V1EnvVar (name = "DOCKERHUB_TOKEN" , value = "{{ ti.xcom_pull(task_ids='Setup', key='dockerhub_token') }}" ),
175+ k8s .V1EnvVar (
176+ name = "DOCKERHUB_USERNAME" , value = "{{ ti.xcom_pull(task_ids='Setup', key='dockerhub_username') }}"
177+ ),
178+ k8s .V1EnvVar (name = "DOCKERHUB_TOKEN" , value = "{{ ti.xcom_pull(task_ids='Setup', key='dockerhub_token') }}" ),
172179 k8s .V1EnvVar (name = "DOCKSTORE_TOKEN" , value = "{{ ti.xcom_pull(task_ids='Setup', key='dockstore_token') }}" ),
173180 k8s .V1EnvVar (name = "GITHUB_REPO" , value = "{{ params.repository }}" ),
174181]
175182
183+
176184def setup (ti = None , ** context ):
177185 """
178186 Task that selects the proper Karpenter Node Pool depending on the user requested resources.
@@ -197,7 +205,7 @@ def setup(ti=None, **context):
197205
198206 required_credentials = ["dockerhub_username" , "dockerhub_token" , "dockstore_token" ]
199207 # make sure all required credentials are provided
200- if ( not set (required_credentials ).issubset (list (credentials_dict .keys () ))):
208+ if not set (required_credentials ).issubset (list (credentials_dict .keys ())):
201209 logging .error (f"Expected all of credentials to run mdps app generator { required_credentials } " )
202210
203211 # use xcom to push to avoid putting credentials to the logs
0 commit comments