test: try this test

Joan Fontanals Martinez · Joan Fontanals Martinez · commit 12a8747ea5e2 · 2023-09-18T13:29:20.000+02:00
Signed-off-by: Joan Fontanals Martinez &lt;joan.martinez@jina.ai&gt;
diff --git a/docarray/utils/create_dynamic_doc_class.py b/docarray/utils/create_dynamic_doc_class.py
@@ -73,9 +73,8 @@ class MyDoc(BaseDoc):
         except TypeError:
             fields[field_name] = (field, field_info)
 
-    validators = model.__validators__ if not is_pydantic_v2 else {'__validators__': model.__pydantic_validator__}
     return create_model(
-        model.__name__, __base__=model, __validators__=validators, **fields
+        model.__name__, __base__=model, **fields
     )
 
 
@@ -208,7 +207,7 @@ def _get_field_annotation_from_schema(
             definitions=definitions,
         )
     elif field_type == 'null':
-        ret = None
+        ret = Type[None]
     else:
         if num_recursions > 0:
             raise ValueError(
@@ -262,9 +261,8 @@ class MyDoc(BaseDoc):
     :param definitions: Parameter used when this method is called recursively to reuse root definitions of other schemas.
     :return: A BaseDoc class dynamically created following the `schema`.
     """
-    print(f'schema {schema}')
     if not definitions:
-        definitions = schema.get('definitions', {})
+        definitions = schema.get('definitions', {}) if not is_pydantic_v2 else schema.get('$defs')
 
     cached_models = cached_models if cached_models is not None else {}
     fields: Dict[str, Any] = {}
@@ -287,11 +285,11 @@ class MyDoc(BaseDoc):
         )
 
     model = create_model(base_doc_name, __base__=BaseDoc, **fields)
-    model.__config__.title = schema.get('title', model.__config__.title)
+    #model.__config__.title = schema.get('title', model.__config__.title)
 
     for k in RESERVED_KEYS:
         if k in schema:
             schema.pop(k)
-    model.__config__.schema_extra = schema
+    #model.__config__.schema_extra = schema
     cached_models[base_doc_name] = model
     return model
diff --git a/tests/units/util/test_create_dynamic_code_class.py b/tests/units/util/test_create_dynamic_code_class.py
@@ -173,7 +173,6 @@ class ResultTestDoc(BaseDoc):
         assert doc.ia == f'ID {i}'
 
 
-@pytest.mark.skipif(is_pydantic_v2, reason="Not working with pydantic v2 for now")
 @pytest.mark.parametrize('transformation', ['proto', 'json'])
 def test_create_empty_doc_list_from_schema(transformation):
     class CustomDoc(BaseDoc):
@@ -259,7 +258,6 @@ class ResultTestDoc(BaseDoc):
     assert len(custom_da) == 0
 
 
-@pytest.mark.skipif(is_pydantic_v2, reason="Not working with pydantic v2 for now")
 def test_create_with_field_info():
     class CustomDoc(BaseDoc):
         """Here I have the description of the class"""
