Commit 30e85bdf authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent 54040bae
Loading
Loading
Loading
Loading
+21 −24
Original line number Diff line number Diff line
@@ -16,7 +16,7 @@
  termcolor,
  yapf,

  # checks
  # tests
  bitsandbytes,
  coverage,
  dvclive,
@@ -26,18 +26,19 @@
  parameterized,
  pytestCheckHook,
  transformers,
  writableTmpDirAsHomeHook,
}:

buildPythonPackage rec {
  pname = "mmengine";
  version = "0.10.6";
  version = "0.10.7";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "open-mmlab";
    repo = "mmengine";
    tag = "v${version}";
    hash = "sha256-J9p+JCtNoBlBvvv4p57/DHUIifYs/jdo+pK+paD+iXI=";
    hash = "sha256-hQnwenuxHQwl+DwQXbIfsKlJkmcRvcHV1roK7q2X1KA=";
  };

  build-system = [ setuptools ];
@@ -53,6 +54,8 @@ buildPythonPackage rec {
    yapf
  ];

  pythonImportsCheck = [ "mmengine" ];

  nativeCheckInputs = [
    bitsandbytes
    coverage
@@ -63,49 +66,43 @@ buildPythonPackage rec {
    parameterized
    pytestCheckHook
    transformers
    writableTmpDirAsHomeHook
  ];

  preCheck =
    ''
      export HOME=$(mktemp -d)
    ''
    # 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" ];
  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"

  disabledTestPaths = [
    # AttributeError
    "tests/test_fileio/test_backends/test_petrel_backend.py"
    # Freezes forever?
    "tests/test_runner/test_activation_checkpointing.py"
    # missing dependencies
    "tests/test_visualizer/test_vis_backend.py"
    # AttributeError: type object 'MagicMock' has no attribute ...
    "--deselect tests/test_fileio/test_backends/test_petrel_backend.py::TestPetrelBackend"
  ];

  disabledTests = [
    # Tests are disabled due to sandbox
    # Require network access
    "test_fileclient"
    "test_http_backend"
    "test_misc"

    # RuntimeError
    "test_dump"
    "test_deepcopy"
    "test_copy"
    "test_lazy_import"
    # AssertionError

    # AssertionError: os is not <module 'os' (frozen)>
    "test_lazy_module"
    # Require unpackaged aim
    "test_experiment"
    "test_add_config"
    "test_add_image"
    "test_add_scalar"
    "test_add_scalars"
    "test_close"
  ];

  meta = {