Fix for #459 (pydantic code gen only) (#460)

This commit is contained in:
Samuel Yvon
2023-02-21 14:41:32 -05:00
committed by GitHub
parent 13d656587c
commit 2fa0be2141
2 changed files with 19 additions and 3 deletions

View File

@@ -7,7 +7,11 @@ import {{ i }}
{% endfor %}
{% if output_file.pydantic_dataclasses %}
from pydantic.dataclasses import dataclass
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from dataclasses import dataclass
else:
from pydantic.dataclasses import dataclass
{%- else -%}
from dataclasses import dataclass
{% endif %}