Skip to content

Commit 00cb30d

Browse files
committed
update spacy tests
1 parent 3dee722 commit 00cb30d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vetiver/tests/test_spacy.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def test_good_prototype_shape(data, spacy_model):
6868
model_schema = v.prototype.model_json_schema()
6969
expected = {
7070
"properties": {
71-
"col": {"examples": "[1]", "title": "Col", "type": "string"},
71+
"col": {"examples": ["1"], "title": "Col", "type": "string"},
7272
},
7373
"required": ["col"],
7474
"title": "prototype",
@@ -77,7 +77,7 @@ def test_good_prototype_shape(data, spacy_model):
7777
except AttributeError: # pydantic v1
7878
model_schema = v.prototype.schema_json()
7979
expected = '{"title": "prototype", "type": "object", "properties": \
80-
{"col": {"title": "Col", "examples": "[1]", "type": "string"}}, "required": ["col"]}'
80+
{"col": {"title": "Col", "examples": ["1"], "type": "string"}}, "required": ["col"]}'
8181

8282
assert model_schema == expected
8383

0 commit comments

Comments
 (0)