python311Packages.mmengine: format with nixfmt

This commit is contained in:
Fabian Affolter 2024-04-05 11:31:42 +02:00
parent b3e11c88c3
commit 42c1fe7977
1 changed files with 32 additions and 34 deletions

View File

@ -1,22 +1,23 @@
{ lib
, addict
, buildPythonPackage
, coverage
, fetchFromGitHub
, lmdb
, matplotlib
, mlflow
, numpy
, opencv4
, parameterized
, pytestCheckHook
, pythonOlder
, pyyaml
, rich
, setuptools
, termcolor
, torch
, yapf
{
lib,
addict,
buildPythonPackage,
coverage,
fetchFromGitHub,
lmdb,
matplotlib,
mlflow,
numpy,
opencv4,
parameterized,
pytestCheckHook,
pythonOlder,
pyyaml,
rich,
setuptools,
termcolor,
torch,
yapf,
}:
buildPythonPackage rec {
@ -33,9 +34,7 @@ buildPythonPackage rec {
hash = "sha256-fKtPDdeKB3vX2mD+Tsicq8KOkPDSACzKK1XLyugdPQ4=";
};
build-system = [
setuptools
];
build-system = [ setuptools ];
dependencies = [
addict
@ -57,19 +56,18 @@ buildPythonPackage rec {
torch
];
preCheck = ''
export HOME=$TMPDIR
''
# Otherwise, the backprop hangs forever. More precisely, this exact line:
# https://github.com/open-mmlab/mmengine/blob/02f80e8bdd38f6713e04a872304861b02157905a/tests/test_runner/test_activation_checkpointing.py#L46
# Solution suggested in https://github.com/pytorch/pytorch/issues/91547#issuecomment-1370011188
+ ''
export MKL_NUM_THREADS=1
'';
preCheck =
''
export HOME=$TMPDIR
''
# Otherwise, the backprop hangs forever. More precisely, this exact line:
# https://github.com/open-mmlab/mmengine/blob/02f80e8bdd38f6713e04a872304861b02157905a/tests/test_runner/test_activation_checkpointing.py#L46
# Solution suggested in https://github.com/pytorch/pytorch/issues/91547#issuecomment-1370011188
+ ''
export MKL_NUM_THREADS=1
'';
pythonImportsCheck = [
"mmengine"
];
pythonImportsCheck = [ "mmengine" ];
disabledTestPaths = [
# AttributeError