hot-fix: incorrect bucket bug (#82)

AllenNeuralDynamics · Mar 12, 2024 · 77ad35d · 77ad35d
1 parent 495e39a
commit 77ad35d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/aind_data_transfer_service/server.py b/src/aind_data_transfer_service/server.py
@@ -183,6 +183,13 @@ async def submit_hpc_jobs(request: Request):  # noqa: C901
                     job["upload_job_settings"]
                 ).s3_prefix
             upload_job_configs = json.loads(job["upload_job_settings"])
+            # This will set the bucket to the private data one
+            if upload_job_configs.get("s3_bucket") is not None:
+                upload_job_configs = json.loads(
+                    BasicUploadJobConfigs.model_validate(
+                        upload_job_configs
+                    ).model_dump_json()
+                )
             # The aws creds to use are different for aind-open-data and
             # everything else
             if upload_job_configs.get("s3_bucket") == OPEN_DATA_BUCKET_NAME: