correct how params are passed for dags

digital-land · Aug 15, 2024 · 5b42b71 · 5b42b71
1 parent f30e8ef
commit 5b42b71
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/dags/collection_generator.py b/dags/collection_generator.py
@@ -42,22 +42,21 @@
             "memory": Param(32768, type="integer"),
             "timeout": Param(480, type="integer")
         },
-        render_template_as_native_obj=True
     ) as dag:
         print(f'running with params:{dag.params}')
         EcsRunTaskOperator(
             task_id=f"{collection}-collection",
             dag=dag,
-            execution_timeout=timedelta(minutes= "{{ params.timeout }}"),
+            execution_timeout=timedelta(minutes= dag.params['timeout']),
             cluster=cluster_name,
             task_definition="development-mwaa-collection-task",
             launch_type="FARGATE",
             overrides={
                 "containerOverrides": [
                     {
                         "name": "development-mwaa-collection-task",
-                        'cpu': "{{ params.cpu }}", 
-                        'memory': "{{ params.memory }}", 
+                        'cpu': dag.params['cpu'], 
+                        'memory': dag.params['memory'], 
                         "environment": [
                             {"name": "COLLECTION_NAME", "value": collection}
                         ],