Commit 11d83ca0 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.compressed-tensors: 0.14.0.1 -> 0.15.0.1 

Diff: https://github.com/neuralmagic/compressed-tensors/compare/0.14.0.1...0.15.0.1

Changelog: https://github.com/neuralmagic/compressed-tensors/releases/tag/0.15.0.1
parent 6cb7aaac

pkgs/development/python-modules/compressed-tensors/default.nix

+10 −8
Original line number Diff line number Diff line
@@ -3,7 +3,6 @@ 
  stdenv,

  buildPythonPackage,

  fetchFromGitHub,

  pythonAtLeast,



  # build-system

  setuptools,
@@ -25,7 +24,7 @@ 


buildPythonPackage (finalAttrs: {

  pname = "compressed-tensors";

  version = "0.14.0.1";

  version = "0.15.0.1";

  pyproject = true;



  # Release on PyPI is missing the `utils` directory, which `setup.py` wants to import
@@ -33,7 +32,7 @@ buildPythonPackage (finalAttrs: { 
    owner = "neuralmagic";

    repo = "compressed-tensors";

    tag = finalAttrs.version;

    hash = "sha256-bncglC6qRE4D6rjoNOXRKh+7ZMSTc9zc3NVzA+NaXuE=";

    hash = "sha256-iiYo3Vne2CYlj+wMHxQFTTU7gb8oNwPtCe873nX5KgA=";

  };



  postPatch = ''
@@ -46,7 +45,9 @@ buildPythonPackage (finalAttrs: { 
    setuptools-scm

  ];



  buildInputs = lib.optional stdenv.cc.isClang llvmPackages.openmp;

  buildInputs = lib.optionals stdenv.cc.isClang [

    llvmPackages.openmp

  ];



  dependencies = [

    frozendict
@@ -56,10 +57,6 @@ buildPythonPackage (finalAttrs: { 
    transformers

  ];



  pythonRelaxDeps = [ "transformers" ];



  doCheck = true;



  pythonImportsCheck = [ "compressed_tensors" ];



  nativeCheckInputs = [
@@ -89,6 +86,11 @@ buildPythonPackage (finalAttrs: { 
    "test_set_item"

    "test_set_item_buffers"

    "test_update_offload_parameter_with_grad"

  ]

  ++ lib.optionals stdenv.hostPlatform.isDarwin [

    # torch._inductor.exc.InductorError: ImportError: dlopen(/nix/var/nix/builds/nix-25002-542173852/torchinductor__nixbld1/xo/cxovsevcfanmw7lgoddbnyhoxes3nzlu7ecugxedaq2zr4f6b2qh.main.so, 0x0002):

    # symbol not found in flat namespace '___kmpc_barrier'

    "test_compress_decompress_module"

  ];



  disabledTestPaths = [