python3Packages.torchmetrics: work around tests duplicate package issue

, pytestCheckHook

, torchmetrics

, pytorch-lightning

, pytest-doctestplus

, pytest-xdist

}:

let

in

buildPythonPackage {

  inherit pname version;

  pyproject = true;

  src = fetchFromGitHub {

    owner = "PyTorchLightning";

    repo = "metrics";

    owner = "Lightning-AI";

    repo = "torchmetrics";

    rev = "refs/tags/v${version}";

    hash = "sha256-g5JuTbiRd8yWx2nM3UE8ejOhuZ0XpAQdS5AC9AlrSFY=";

  };

    cloudpickle

    psutil

    pytestCheckHook

    pytest-doctestplus

    pytest-xdist

  ];

  # A cyclic dependency in: integrations/test_lightning.py

  doCheck = false;

  passthru.tests.check = torchmetrics.overridePythonAttrs (_: {

    pname = "${pname}-check";

    doCheck = true;

    # We don't have to install because the only purpose

    # of this passthru test is to, well, test.

    # This fixes having to set `catchConflicts` to false.

    dontInstall = true;

  });

  disabledTests = [

    # `IndexError: list index out of range`

    "test_metric_lightning_log"

  ];

  disabledTestPaths = [

    # These require too many "leftpad-level" dependencies

    "tests/text"

    "tests/audio"

    "tests/image"

    # Also too cross-dependent

    "tests/unittests"

    # A few non-deterministic things like test_check_compute_groups_is_faster

    "tests/bases/test_collections.py"

    # A trillion import path mismatch errors

    "src/torchmetrics"

  ];

  pythonImportsCheck = [

  meta = with lib; {

    description = "Machine learning metrics for distributed, scalable PyTorch applications (used in pytorch-lightning)";

    homepage = "https://torchmetrics.readthedocs.io";

    homepage = "https://lightning.ai/docs/torchmetrics/";

    license = licenses.asl20;

    maintainers = with maintainers; [

      SomeoneSerge

    ];

  };

}