diff --git a/pkgs/development/python-modules/mmengine/default.nix b/pkgs/development/python-modules/mmengine/default.nix index 347d22d569e0..a59db4926b8f 100644 --- a/pkgs/development/python-modules/mmengine/default.nix +++ b/pkgs/development/python-modules/mmengine/default.nix @@ -13,6 +13,7 @@ , pythonOlder , pyyaml , rich +, setuptools , termcolor , torch , yapf @@ -32,7 +33,11 @@ buildPythonPackage rec { hash = "sha256-fKtPDdeKB3vX2mD+Tsicq8KOkPDSACzKK1XLyugdPQ4="; }; - propagatedBuildInputs = [ + build-system = [ + setuptools + ]; + + dependencies = [ addict matplotlib numpy @@ -47,9 +52,9 @@ buildPythonPackage rec { coverage lmdb mlflow - torch parameterized pytestCheckHook + torch ]; preCheck = '' @@ -87,7 +92,6 @@ buildPythonPackage rec { "test_lazy_import" # AssertionError "test_lazy_module" - # Require unpackaged aim "test_experiment" "test_add_config" @@ -97,6 +101,11 @@ buildPythonPackage rec { "test_close" ]; + pytestFlagsArray = [ + "-W" + "ignore::pytest.PytestRemovedIn8Warning" + ]; + meta = with lib; { description = "Library for training deep learning models based on PyTorch"; homepage = "https://github.com/open-mmlab/mmengine";