Unverified Commit 3765ba2a authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python312Packages.pytorch-metric-learning: cleanup, fix on darwin (#410143)

parents 980ae985 b1a415a4

pkgs/development/python-modules/pytorch-metric-learning/default.nix

+13 −8
Original line number Diff line number Diff line 
{

  stdenv,

  lib,

  stdenv,

  config,

  buildPythonPackage,

  fetchFromGitHub,

  isPy27,

  config,



  # build-system

  setuptools,
@@ -20,9 +19,11 @@ 
  tensorboard,



  # tests

  cudaSupport ? config.cudaSupport,

  pytestCheckHook,

  torchvision,

  writableTmpDirAsHomeHook,



  cudaSupport ? config.cudaSupport,

}:



buildPythonPackage rec {
@@ -30,11 +31,9 @@ buildPythonPackage rec { 
  version = "2.8.1";

  pyproject = true;



  disabled = isPy27;



  src = fetchFromGitHub {

    owner = "KevinMusgrave";

    repo = pname;

    repo = "pytorch-metric-learning";

    tag = "v${version}";

    hash = "sha256-WO/gv8rKkxY3pR627WrEPVyvZnvUZIKMzOierIW8bJA=";

  };
@@ -64,7 +63,6 @@ buildPythonPackage rec { 
  };



  preCheck = ''

    export HOME=$TMP

    export TEST_DEVICE=cpu

    export TEST_DTYPES=float32,float64  # half-precision tests fail on CPU

  '';
@@ -73,6 +71,7 @@ buildPythonPackage rec { 
  nativeCheckInputs = [

    pytestCheckHook

    torchvision

    writableTmpDirAsHomeHook

  ] ++ lib.flatten (lib.attrValues optional-dependencies);



  disabledTests =
@@ -100,6 +99,12 @@ buildPythonPackage rec { 
      "test_with_same_parent_label_tester"

    ];



  disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [

    # Fatal Python error: Segmentation fault

    "tests/testers/"

    "tests/utils/"

  ];



  meta = {

    description = "Metric learning library for PyTorch";

    homepage = "https://github.com/KevinMusgrave/pytorch-metric-learning";