From 8abcba5c2189d85fc7a1db0d666922b22f52beac Mon Sep 17 00:00:00 2001
From: Dipika Sikka <dipikasikka1@gmail.com>
Date: Mon, 10 Nov 2025 11:51:29 -0500
Subject: [PATCH] fix

---
 tests/llmcompressor/modifiers/calibration/test_lifecycle.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/llmcompressor/modifiers/calibration/test_lifecycle.py b/tests/llmcompressor/modifiers/calibration/test_lifecycle.py
index 0878975f67..0d72ba897e 100644
--- a/tests/llmcompressor/modifiers/calibration/test_lifecycle.py
+++ b/tests/llmcompressor/modifiers/calibration/test_lifecycle.py
@@ -1,6 +1,7 @@
 import pytest
 import torch
 from compressed_tensors.quantization import (
+    FP8_E4M3_DATA,
     QuantizationScheme,
     forward_quantize,
     initialize_module_for_quantization,
@@ -83,6 +84,7 @@
                 symmetric=True,
                 strategy="tensor_group",  # requires float4
                 group_size=3,
+                scale_dtype=FP8_E4M3_DATA.dtype,
             ),
             torch.tensor([[0, 3], [6, 9], [12, 15], [18, 21]]),
             torch.tensor([[2, 5], [8, 11], [14, 17], [20, 23]]),
@@ -195,6 +197,7 @@ def test_static_weight_quantization(
                 strategy="tensor_group",
                 dynamic="local",
                 group_size=3,
+                scale_dtype=FP8_E4M3_DATA.dtype,
             ),
             None,
             None,