chore: generalize sample parser

aphp · Jun 16, 2024 · 6f37ab4 · 6f37ab4
1 parent 1fd8987
commit 6f37ab4
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/edsnlp/utils/examples.py b/edsnlp/utils/examples.py
@@ -1,8 +1,8 @@
-import json
 import re
 from typing import Any, Dict, List, Tuple, Union
 
 import regex
+from confit.utils.xjson import loads
 from pydantic import BaseModel, validator
 
 
@@ -19,14 +19,7 @@ class Modifier(BaseModel):
 
     @validator("value")
     def optional_dict_parsing(cls, v):
-        if v in ["True", "False"]:
-            return v == "True"
-        if isinstance(v, str):
-            try:
-                return json.loads(v.replace("'", '"'))
-            except json.JSONDecodeError:
-                return v
-        return v
+        return loads(v) if isinstance(v, str) else v
 
 
 class Entity(BaseModel):