fix: improve pydantic v1/v2 support

aphp · Nov 10, 2024 · 3707049 · 3707049
1 parent d01f29a
commit 3707049
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/edsnlp/pipes/ner/adicap/adicap.py b/edsnlp/pipes/ner/adicap/adicap.py
@@ -98,7 +98,7 @@ class AdicapMatcher(ContextualMatcher):
     ent.label_
     # Out: adicap
 
-    ent._.adicap.model_dump()
+    ent._.adicap.dict()
     # Out: {'code': 'BHGS0040',
     # 'sampling_mode': 'BIOPSIE CHIRURGICALE',
     # 'technic': 'HISTOLOGIE ET CYTOLOGIE PAR INCLUSION',

diff --git a/edsnlp/pipes/ner/tnm/model.py b/edsnlp/pipes/ner/tnm/model.py
@@ -197,3 +197,6 @@ def dict(
                 d[k] = v.value
 
         return d
+
+    if pydantic.VERSION < "2":
+        model_dump = pydantic.BaseModel.dict