diff options
| -rw-r--r-- | python-packages/json_schemas/src/zero_ex/json_schemas/__init__.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/python-packages/json_schemas/src/zero_ex/json_schemas/__init__.py b/python-packages/json_schemas/src/zero_ex/json_schemas/__init__.py index f609ea103..1752c5761 100644 --- a/python-packages/json_schemas/src/zero_ex/json_schemas/__init__.py +++ b/python-packages/json_schemas/src/zero_ex/json_schemas/__init__.py @@ -27,7 +27,9 @@ class _LocalRefResolver(jsonschema.RefResolver): # handle weird special cases _ref = _ref.replace("ECSignature", "EcSignature") - _ref = _ref.replace("Schema", "") + if _ref.endswith("Schema"): + # strip off the Schema suffix + _ref = _ref[:-6] return f"{snakecase(_ref)}_schema.json" |