Fix tests

huggingface · Nov 18, 2024 · aeddb25 · aeddb25
1 parent a2bc04e
commit aeddb25
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/services/worker/tests/job_runners/dataset/test_croissant_crumbs.py b/services/worker/tests/job_runners/dataset/test_croissant_crumbs.py
@@ -163,7 +163,8 @@ def test_get_croissant_crumbs_from_dataset_infos() -> None:
     assert len(croissant_crumbs["recordSet"][3]["field"]) == 6
     for field in croissant_crumbs["recordSet"][1]["field"]:
         assert field["@type"] == "cr:Field"
-        assert field["dataType"] == "sc:Text"
+        if "subField" not in field:
+            assert field["dataType"] == "sc:Text"
     assert len(croissant_crumbs["recordSet"][1]["field"]) == len(squad_info["features"]) + 1
 
     # Test distribution.