added preproc task to training DAG

interTwin-eu · Sep 15, 2024 · be1c875 · be1c875
1 parent dcc1abb
commit be1c875
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dags_airflow_ale_trainmod.py b/dags_airflow_ale_trainmod.py
@@ -151,12 +151,12 @@ def check_response(response):
         log_events_on_failure=True,
         # enable xcom
         do_xcom_push=True,
-        volumes = k8s.V1Volume(
+        volumes = [k8s.V1Volume(
         name="gwdatal",
         persistent_volume_claim=k8s.V1PersistentVolumeClaimVolumeSource(claim_name="gwdatal"),
-        ),
-        volume_mounts=k8s.V1VolumeMount(mount_path="/app/data", name="gwdatal", sub_path=None, read_only=False)
-        ,
+        )],
+        volume_mounts=[k8s.V1VolumeMount(mount_path="/app/data", name="gwdatal", sub_path=None, read_only=False)
+        ],
         dag=dag,
 
         #env_vars={"NAME_TO_GREET": f"{name}"},