Merge pull request #789 from deepmodeling/zjgemi

fix: add prefix when upload_s3 with key
deepmodeling · Apr 1, 2024 · be0cb8a · be0cb8a
2 parents 184e3f2 + fed4697
commit be0cb8a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/dflow/utils.py b/src/dflow/utils.py
@@ -395,7 +395,9 @@ def upload_s3(
     else:
         client = MinioClient(**kwargs)
     if key is not None:
-        pass
+        if not key.startswith(s3_config["prefix"]) and not any(
+                [key.startswith(p) for p in s3_config["extra_prefixes"]]):
+            key = s3_config["prefix"] + key
     elif prefix is not None:
         if prefix[-1] != "/":
             prefix += "/"