Fix spacy test failure caused by spacy 3.10 (mlflow#4541)

* remove _sourced_vectors_hashes Signed-off-by: harupy <17039389+harupy@users.noreply.github.com> * set default Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
BLACKBUCK-LABS · Jul 9, 2021 · ec6a16e · ec6a16e
1 parent c96f4d8
commit ec6a16e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/spacy/test_spacy_model_export.py b/tests/spacy/test_spacy_model_export.py
@@ -84,6 +84,10 @@ def test_model_save_load(spacy_model_with_data, model_path):
     mlflow.spacy.save_model(spacy_model=spacy_model, path=model_path)
     loaded_model = mlflow.spacy.load_model(model_path)
 
+    # Remove a `_sourced_vectors_hashes` field which is added when spaCy loads a model:
+    # https://github.com/explosion/spaCy/blob/e8ef4a46d5dbc9bb6d629ecd0b02721d6bdf2f87/spacy/language.py#L1701
+    loaded_model.meta.pop("_sourced_vectors_hashes", None)
+
     # Comparing the meta dictionaries for the original and loaded models
     assert spacy_model.meta == loaded_model.meta