diff --git a/pkgs/development/python-modules/mmengine/default.nix b/pkgs/development/python-modules/mmengine/default.nix index a59db4926b8f..7187fe809875 100644 --- a/pkgs/development/python-modules/mmengine/default.nix +++ b/pkgs/development/python-modules/mmengine/default.nix @@ -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