223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge branch '242-optional-numba-for-yaml-emitter' into 'development' 

yaml.CSafeDumper can be subclassed

Closes #242

See merge request damask/DAMASK!678
This commit is contained in:
Martin Diehl 2022-12-06 20:17:24 +00:00
parent 9cb1d75d05 ba2062e4ed
commit 418e2f022b
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
python/damask/_config.py
View File

@ -8,8 +8,10 @@ import numpy as np
import yaml import yaml
try: try:
from yaml import CSafeLoader as SafeLoader from yaml import CSafeLoader as SafeLoader
from yaml import CSafeDumper as SafeDumper
except ImportError: except ImportError:
from yaml import SafeLoader # type: ignore from yaml import SafeLoader # type: ignore
from yaml import SafeDumper # type: ignore
from ._typehints import FileHandle from ._typehints import FileHandle
from . import Rotation from . import Rotation
@ -17,20 +19,20 @@ from . import util
MyType = TypeVar('MyType', bound='Config') MyType = TypeVar('MyType', bound='Config')
class NiceDumper(yaml.SafeDumper): class NiceDumper(SafeDumper):
"""Make YAML readable for humans.""" """Make YAML readable for humans."""
def write_line_break(self, def write_line_break(self,
data: Optional[str] = None): data: Optional[str] = None):
super().write_line_break(data) super().write_line_break(data) # type: ignore
if len(self.indents) == 1: if len(self.indents) == 1: # type: ignore
super().write_line_break() super().write_line_break() # type: ignore
def increase_indent(self, def increase_indent(self,
flow: bool = False, flow: bool = False,
indentless: bool = False): indentless: bool = False):
return super().increase_indent(flow, False) return super().increase_indent(flow, False) # type: ignore
def represent_data(self, def represent_data(self,
data: Any): data: Any):