Skip to content

Commit fd2875a

Browse files
committed
Fix lint
1 parent c7b07e3 commit fd2875a

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

tests/extract/test_validation.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def some_data() -> t.Iterator[TDataItems]:
261261
def 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

tests/pipeline/test_pipeline_extra.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -863,23 +863,23 @@ def inconsistent_data(dtype: str):
863863
def 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]] = []

0 commit comments

Comments
 (0)