Set env.

digital-land · Jul 23, 2024 · 9638388 · 9638388
1 parent bd5e354
commit 9638388
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/dags/dynamic_data_workflow_dag.py b/dags/dynamic_data_workflow_dag.py
@@ -18,9 +18,9 @@
         {
             "name": "collection-task",
             "image": "collection-task",
-            #"workingDirectory": "/usr/bin",
-            #"entryPoint": ["sh", "-c"],
-            #"command": ["ls"],
+            # "workingDirectory": "/usr/bin",
+            # "entryPoint": ["sh", "-c"],
+            # "command": ["ls"],
             # "logConfiguration": {
             #     "logDriver": "awslogs",
             #     "options": {
@@ -69,13 +69,13 @@
             cluster=cluster_name,
             task_definition=collection_task.output,
             launch_type="FARGATE",
-            overrides={},
-            # overrides={"containerOverrides": [
-            #     {
-            #         "name": "test",
-            #         "command": ["python", "-c", "import time; for i in range(30): print(i); time.sleep(1)"],
-            #     },
-            # ]},
+            overrides={
+                "containerOverrides": [
+                    {
+                        "environment": [{"COLLECTION_NAME": collection}],
+                    },
+                ]
+            },
             network_configuration={
                 "awsvpcConfiguration": {
                     "subnets": ["subnet-05a0d548ea8d901ab", "subnet-07252405b5369afd3"],
@@ -89,7 +89,7 @@
             # awslogs_fetch_interval=timedelta(seconds=5)
         )
 
-    break # Just do the first one for now
+    break  # Just do the first one for now
 
     """
     with DAG(
@@ -130,4 +130,4 @@
             # awslogs_stream_prefix=f"ecs/test",
             awslogs_fetch_interval=timedelta(seconds=5),
         )
-    """
+    """