Merge pull request deepmodeling#68 from dptech-corp/zjgemi

fix: handle big parameter whose name changes
njzjz · Jun 14, 2022 · 671cf94 · 671cf94
2 parents 685eba5 + 7513567
commit 671cf94
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # CHANGELOG
 
+## 1.0.19
+
+### Fix
+
+- fix: handle big parameter whose name changes
+
 ## 1.0.18
 
 ### Added

diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-1.0.18
+1.0.19
diff --git a/src/dflow/argo_objects.py b/src/dflow/argo_objects.py
@@ -74,7 +74,9 @@ def handle_big_parameters(self, io):
                         io.parameters = {}
                     with tempfile.TemporaryDirectory() as tmpdir:
                         download_artifact(art, path=tmpdir)
-                        with open(os.path.join(tmpdir, name[13:]), "r") as f:
+                        fs = os.listdir(tmpdir)
+                        assert len(fs) == 1
+                        with open(os.path.join(tmpdir, fs[0]), "r") as f:
                             content = jsonpickle.loads(f.read())
                             param = {"name": name[13:], "save_as_artifact": True}
                             if "type" in content: