[tool.pytest.ini_options] | |
markers = [ | |
"base: marker for model tests using the basic setup", | |
"cfg: marker for model tests checking the config", | |
"torchscript: marker for model tests using torchscript", | |
"features: marker for model tests checking feature extraction", | |
"fxforward: marker for model tests using torch fx (only forward)", | |
"fxbackward: marker for model tests using torch fx (only backward)", | |
] | |
[tool.black] | |
line-length = 120 | |
target-version = ['py37', 'py38', 'py39', 'py310', 'py311'] | |
skip-string-normalization = true | |