properly handle gcs folders - dags prefix is part of blob name not bucket name

leahecole · leahecole · commit 7e01979bb373 · 2021-12-14T15:47:07.000-08:00
diff --git a/composer/cicd_sample/utils/add_dags_to_composer.py b/composer/cicd_sample/utils/add_dags_to_composer.py
@@ -51,7 +51,7 @@ def upload_dags_to_composer(dags_directory: str, bucket_name: str) -> None:
 
         for dag in dags:
             # Remove path to temp dir
-            dag = dag.replace(f"{temp_dir}/", "")
+            dag = dag.replace(f"{temp_dir}/", "dags/")
             # Upload to your bucket
             blob = bucket.blob(dag)
             blob.upload_from_string(dag)
