File tree Expand file tree Collapse file tree 2 files changed +10
-10
lines changed
Expand file tree Collapse file tree 2 files changed +10
-10
lines changed Original file line number Diff line number Diff line change @@ -261,7 +261,7 @@ def some_data() -> t.Iterator[TDataItems]:
261261def test_pydantic_validator_return_models (return_models , yield_models , yield_list ):
262262
263263 class TempModel (BaseModel ):
264- id : int
264+ _id : int
265265 dlt_config : t .ClassVar [DltConfig ] = {"return_validated_models" : return_models }
266266
267267 validator = PydanticValidator (
@@ -276,14 +276,14 @@ class TempModel(BaseModel):
276276 item : t .Any
277277 if yield_list :
278278 if yield_models :
279- item = [VM (id = i ) for i in range (3 )]
279+ item = [VM (_id = i ) for i in range (3 )]
280280 else :
281- item = [{"id " : i } for i in range (3 )]
281+ item = [{"_id " : i } for i in range (3 )]
282282 else :
283283 if yield_models :
284- item = VM (id = 1 )
284+ item = VM (_id = 1 )
285285 else :
286- item = {"id " : 1 }
286+ item = {"_id " : 1 }
287287
288288 validated = validator (item )
289289
Original file line number Diff line number Diff line change @@ -863,23 +863,23 @@ def inconsistent_data(dtype: str):
863863def test_pydantic_return_validated_models (return_models , yield_models , yield_list ):
864864
865865 class TempModel (BaseModel ):
866- id : int
866+ _id : int
867867 dlt_config : ClassVar [DltConfig ] = {"return_validated_models" : return_models }
868868
869869 @dlt .resource (columns = TempModel )
870870 def data ():
871871 if yield_list :
872872 if yield_models :
873- yield [TempModel (id = i ) for i in range (3 )]
873+ yield [TempModel (_id = i ) for i in range (3 )]
874874 else :
875- yield [{"id " : i } for i in range (3 )]
875+ yield [{"_id " : i } for i in range (3 )]
876876 else :
877877 if yield_models :
878878 for i in range (3 ):
879- yield TempModel (id = i )
879+ yield TempModel (_id = i )
880880 else :
881881 for i in range (3 ):
882- yield {"id " : i }
882+ yield {"_id " : i }
883883
884884 seen_outer : list [type ] = []
885885 seen_inner : list [Union [type , None ]] = []
You can’t perform that action at this time.
0 commit comments