From ee57b87f5f32c9dd4ef176bef110fa9009683197 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 24 Aug 2025 01:14:31 +0200
Subject: [PATCH] python3Packages.compressed-tensors: 0.10.2 -> 0.11.0

Diff: https://github.com/neuralmagic/compressed-tensors/compare/0.10.2...0.11.0

Changelog: https://github.com/neuralmagic/compressed-tensors/releases/tag/0.11.0
---
 .../compressed-tensors/default.nix            | 33 +++++++++++++++----
 1 file changed, 27 insertions(+), 6 deletions(-)

diff --git a/pkgs/development/python-modules/compressed-tensors/default.nix b/pkgs/development/python-modules/compressed-tensors/default.nix
index 4c5eea81b0b6..3c6fd45b0ebc 100644
--- a/pkgs/development/python-modules/compressed-tensors/default.nix
+++ b/pkgs/development/python-modules/compressed-tensors/default.nix
@@ -1,12 +1,19 @@
 {
   lib,
-  stdenv,
   buildPythonPackage,
   fetchFromGitHub,
+
+  # build-system
   setuptools,
+  setuptools-scm,
+
+  # dependencies
+  frozendict,
   pydantic,
   torch,
   transformers,
+
+  # tests
   nbconvert,
   nbformat,
   pytestCheckHook,
@@ -14,7 +21,7 @@
 
 buildPythonPackage rec {
   pname = "compressed-tensors";
-  version = "0.10.2";
+  version = "0.11.0";
   pyproject = true;
 
   # Release on PyPI is missing the `utils` directory, which `setup.py` wants to import
@@ -22,12 +29,21 @@ buildPythonPackage rec {
     owner = "neuralmagic";
     repo = "compressed-tensors";
     tag = version;
-    hash = "sha256-BJsMyCs+rupt5+i5JlO7oY08Udc8hI3ZnMiN+8ja0mc=";
+    hash = "sha256-sSXn4/N/Pn+wOCY1Z0ziqFxfMRvRA1c90jPOBe+SwZw=";
   };
 
-  build-system = [ setuptools ];
+  postPatch = ''
+    substituteInPlace pyproject.toml \
+      --replace-fail "setuptools_scm==8.2.0" "setuptools_scm"
+  '';
+
+  build-system = [
+    setuptools
+    setuptools-scm
+  ];
 
   dependencies = [
+    frozendict
     pydantic
     torch
     transformers
@@ -45,12 +61,17 @@ buildPythonPackage rec {
 
   disabledTests = [
     # these try to download models from HF Hub
+    "test_apply_tinyllama_dynamic_activations"
+    "test_compress_model"
+    "test_compress_model_meta"
+    "test_compressed_linear_from_linear_usage"
+    "test_decompress_model"
     "test_get_observer_token_count"
     "test_kv_cache_quantization"
-    "test_target_prioritization"
     "test_load_compressed_sharded"
+    "test_model_forward_pass"
     "test_save_compressed_model"
-    "test_apply_tinyllama_dynamic_activations"
+    "test_target_prioritization"
   ];
 
   disabledTestPaths = [