Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.9 kB · 126 lines
Nix
at flake-libs
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127{ lib, buildPythonPackage, fetchFromGitHub, fetchpatch,
# build-system setuptools,
# dependencies addict, matplotlib, numpy, opencv4, pyyaml, rich, termcolor, yapf,
# tests bitsandbytes, coverage, dvclive, lion-pytorch, lmdb, mlflow, parameterized, pytestCheckHook, transformers, writableTmpDirAsHomeHook,}:
buildPythonPackage rec { pname = "mmengine"; version = "0.10.7"; pyproject = true;
src = fetchFromGitHub { owner = "open-mmlab"; repo = "mmengine"; tag = "v${version}"; hash = "sha256-hQnwenuxHQwl+DwQXbIfsKlJkmcRvcHV1roK7q2X1KA="; };
patches = [ # Explicitly disable weights_only in torch.load calls # https://github.com/open-mmlab/mmengine/pull/1650 (fetchpatch { name = "torch-2.6.0-compat.patch"; url = "https://github.com/open-mmlab/mmengine/pull/1650/commits/c21b8431b2c625560a3866c65328cff0380ba1f8.patch"; hash = "sha256-SLr030IdYD9wM/jPJuZd+Dr1jjFx/5/YkJj/IwhnNQg="; }) ];
build-system = [ setuptools ];
dependencies = [ addict matplotlib numpy opencv4 pyyaml rich termcolor yapf ];
pythonImportsCheck = [ "mmengine" ];
nativeCheckInputs = [ bitsandbytes coverage dvclive lion-pytorch lmdb mlflow parameterized pytestCheckHook transformers writableTmpDirAsHomeHook ];
preCheck = # 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 '';
pytestFlagsArray = [ # Require unpackaged aim "--deselect tests/test_visualizer/test_vis_backend.py::TestAimVisBackend"
# Cannot find SSL certificate # _pygit2.GitError: OpenSSL error: failed to load certificates: error:00000000:lib(0)::reason(0) "--deselect tests/test_visualizer/test_vis_backend.py::TestDVCLiveVisBackend"
# AttributeError: type object 'MagicMock' has no attribute ... "--deselect tests/test_fileio/test_backends/test_petrel_backend.py::TestPetrelBackend" ];
disabledTests = [ # Require network access "test_fileclient" "test_http_backend" "test_misc"
# RuntimeError "test_dump" "test_deepcopy" "test_copy" "test_lazy_import"
# AssertionError: os is not <module 'os' (frozen)> "test_lazy_module" ];
meta = { description = "Library for training deep learning models based on PyTorch"; homepage = "https://github.com/open-mmlab/mmengine"; changelog = "https://github.com/open-mmlab/mmengine/releases/tag/v${version}"; license = with lib.licenses; [ asl20 ]; maintainers = with lib.maintainers; [ rxiao ]; };}