Skip to content

Commit dfec94d

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 1852aa9 commit dfec94d

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

src/datamodel_code_generator/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@
4848

4949
g_extend_yaml_scientifc_notation: bool = False
5050

51+
5152
def get_safeloader_wrapper(safeloader: type[SafeLoader]) -> type[SafeLoader]:
5253
return get_safeloader(safeloader, extend_yaml_scientifc_notation=g_extend_yaml_scientifc_notation)
5354

src/datamodel_code_generator/util.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,15 +16,16 @@
1616
from pathlib import Path
1717
from typing import Literal
1818

19+
from yaml import CSafeLoader as SafeLoader
1920
from yaml import SafeLoader
2021

2122
def load_toml(path: Path) -> dict[str, Any]: ...
2223

2324
else:
2425
try:
25-
from yaml import CSafeLoader as SafeLoader
26+
pass
2627
except ImportError: # pragma: no cover
27-
from yaml import SafeLoader
28+
pass
2829

2930
try:
3031
from tomllib import load as load_tomllib
@@ -35,6 +36,7 @@ def load_toml(path: Path) -> dict[str, Any]:
3536
with path.open("rb") as f:
3637
return load_tomllib(f)
3738

39+
3840
def get_safeloader(safeloader: type[SafeLoader], *, extend_yaml_scientifc_notation: bool) -> type[SafeLoader]:
3941
safeloadertemp = copy.deepcopy(safeloader)
4042
safeloadertemp.yaml_constructors = copy.deepcopy(safeloader.yaml_constructors)

0 commit comments

Comments
 (0)