From 57b0c5069e289885478e63fdf64b2ce0f7328967 Mon Sep 17 00:00:00 2001
From: Sarah Clark <seclark@nextquestion.net>
Date: Tue, 24 Feb 2026 12:18:31 -0800
Subject: [PATCH] python3Packages.compressed-tensors: fix build

add missing dependency, disable broken tests
---
 .../python-modules/compressed-tensors/default.nix    | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/pkgs/development/python-modules/compressed-tensors/default.nix b/pkgs/development/python-modules/compressed-tensors/default.nix
index c2c93147d64a..08c0b149d3ce 100644
--- a/pkgs/development/python-modules/compressed-tensors/default.nix
+++ b/pkgs/development/python-modules/compressed-tensors/default.nix
@@ -1,14 +1,17 @@
 {
   lib,
+  stdenv,
   buildPythonPackage,
   fetchFromGitHub,
 
   # build-system
   setuptools,
   setuptools-scm,
+  llvmPackages,
 
   # dependencies
   frozendict,
+  loguru,
   pydantic,
   torch,
   transformers,
@@ -42,8 +45,11 @@ buildPythonPackage rec {
     setuptools-scm
   ];
 
+  buildInputs = lib.optional stdenv.cc.isClang llvmPackages.openmp;
+
   dependencies = [
     frozendict
+    loguru
     pydantic
     torch
     transformers
@@ -72,11 +78,17 @@ buildPythonPackage rec {
     "test_model_forward_pass"
     "test_save_compressed_model"
     "test_target_prioritization"
+    "test_expand_targets_with_llama_stories"
   ];
 
   disabledTestPaths = [
     # these try to download models from HF Hub
     "tests/test_quantization/lifecycle/test_apply.py"
+    # RuntimeError: The weights trying to be saved contained shared tensors
+    "tests/test_transform/factory/test_serialization.py::test_serialization[True-hadamard]"
+    "tests/test_transform/factory/test_serialization.py::test_serialization[True-random-hadamard]"
+    "tests/test_transform/factory/test_serialization.py::test_serialization[False-hadamard]"
+    "tests/test_transform/factory/test_serialization.py::test_serialization[False-random-hadamard]"
   ];
 
   meta = {