fix(api): rename to avoid shadowing type

ssube · Jan 11, 2023 · 48f42e5 · 48f42e5
1 parent 07fa81a
commit 48f42e5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/api/onnx_web/serve.py b/api/onnx_web/serve.py
@@ -148,9 +148,9 @@ def json_with_cors(data, origin='*'):
     return res
 
 
-def make_output_path(type: str, seed: int, params: Tuple[Union[str, int, float]]):
+def make_output_path(mode: str, seed: int, params: Tuple[Union[str, int, float]]):
     sha = sha256()
-    sha.update(type.encode('utf-8'))
+    sha.update(mode.encode('utf-8'))
     for param in params:
         if isinstance(param, str):
             sha.update(param.encode('utf-8'))
@@ -161,7 +161,7 @@ def make_output_path(type: str, seed: int, params: Tuple[Union[str, int, float]]
         else:
             print('cannot hash param: %s, %s' % (param, type(param)))
 
-    output_file = '%s_%s_%s.png' % (type, seed, sha.hexdigest())
+    output_file = '%s_%s_%s.png' % (mode, seed, sha.hexdigest())
     output_full = safer_join(output_path, output_file)
 
     return (output_file, output_full)

diff --git a/onnx-web.code-workspace b/onnx-web.code-workspace
@@ -30,6 +30,7 @@
 			"numpy",
 			"Onnx",
 			"onnxruntime",
+			"outpaint",
 			"pndm",
 			"pretrained",
 			"protobuf",