fix comment

neuralmagic · Apr 12, 2024 · b5a07c4 · b5a07c4
1 parent 24e04b6
commit b5a07c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/quantization/lifecycle/test_apply.py b/tests/quantization/lifecycle/test_apply.py
@@ -57,9 +57,11 @@ def _test_layer_quantization_status(module, inputs: bool, weights: bool):
     assert hasattr(module, "quantization_scheme") == quantized
     assert hasattr(module, "quantization_status") == quantized
 
-    # check for inputs
+    # check inputs matches expected
     assert hasattr(module, "input_scale") == inputs
     assert hasattr(module, "input_zero_point") == inputs
+
+    # check weights matches expected
     assert hasattr(module, "weight_scale") == weights
     assert hasattr(module, "weight_zero_point") == weights