diff --git a/pkgs/development/python-modules/compressed-tensors/default.nix b/pkgs/development/python-modules/compressed-tensors/default.nix index 1dd61d2e9498..9134ea3380e6 100644 --- a/pkgs/development/python-modules/compressed-tensors/default.nix +++ b/pkgs/development/python-modules/compressed-tensors/default.nix @@ -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 = [