Skip to content

Commit 7047cae

Browse files
committed
fix serialize_utils bugs
1 parent 6441ef8 commit 7047cae

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/emd/models/utils/serialize_utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,15 @@
22
import argparse
33

44
JSON_DOUBLE_QUOTE_REPLACE = '<!>'
5+
JSON_SINGLE_QUOTE_REPLACE = '<*>'
56

67

78
def load_extra_params(string):
8-
string = string.replace(JSON_DOUBLE_QUOTE_REPLACE,'"')
9+
string = string.replace(JSON_DOUBLE_QUOTE_REPLACE,'"').replace(JSON_SINGLE_QUOTE_REPLACE,"'")
910
try:
1011
return json.loads(string)
1112
except json.JSONDecodeError:
1213
raise argparse.ArgumentTypeError(f"Invalid dictionary format: {string}")
1314

1415
def dump_extra_params(d:dict):
15-
return json.dumps(d).replace('"', JSON_DOUBLE_QUOTE_REPLACE)
16+
return json.dumps(d).replace("'", JSON_SINGLE_QUOTE_REPLACE).replace('"', JSON_DOUBLE_QUOTE_REPLACE)

0 commit comments

Comments
 (0)