Skip to content

Commit d2d9abe

Browse files
bkalashnikovbogdandm
bkalashnikov
authored andcommitted
Fix pydantic generation
1 parent 4596d29 commit d2d9abe

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/test_code_generation/test_pydantic_generation.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353
]
5454
},
5555
"generated": trim(f"""
56-
from pydantic import BaseModel, Field
56+
from pydantic.v1 import BaseModel, Field
5757
5858
5959
class Test(BaseModel):
@@ -120,7 +120,7 @@ class Test(BaseModel):
120120
}
121121
},
122122
"generated": trim(f"""
123-
from pydantic import BaseModel, Field
123+
from pydantic.v1 import BaseModel, Field
124124
from typing import Dict, List, Optional
125125
126126
@@ -146,7 +146,7 @@ class Test(BaseModel):
146146
"u": DUnion(DDict(IntString), DList(DList(IntString))),
147147
}),
148148
"generated": trim("""
149-
from pydantic import BaseModel, Field
149+
from pydantic.v1 import BaseModel, Field
150150
from typing import Dict, List, Optional, Union
151151
152152

0 commit comments

Comments
 (0)