Skip to content

Commit 4ad5c49

Browse files
committed
fix quality
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 8973328 commit 4ad5c49

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_quantization/lifecycle/test_static_lifecycle.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,7 @@ def calibrate_input_hook(_, args):
250250
global_scale = linear.input_observer.get_global_scale(args[0])
251251
linear.input_global_scale.data = global_scale
252252

253-
if linear.quantization_scheme.input_activations.dynamic == False:
253+
if linear.quantization_scheme.input_activations.dynamic is False:
254254
scale, zero_point = linear.input_observer(args[0])
255255
linear.input_scale.data = scale
256256
linear.input_zero_point.data = zero_point
@@ -331,7 +331,7 @@ def test_static_attention_quantization(
331331
attention.k_observer = MockMinMaxObserver(attention, base_name="k")
332332

333333
# calibrate quantization parameters
334-
if scheme.input_activations.dynamic == False:
334+
if scheme.input_activations.dynamic is False:
335335
scale, zero_point = attention.k_observer(input)
336336
attention.k_scale.data = scale
337337
attention.k_zero_point.data = zero_point

0 commit comments

Comments
 (0)