diff --git a/tests/python/relay/test_pass_auto_quantize.py b/tests/python/relay/test_pass_auto_quantize.py index 9a19052def48e..e357f80d0f520 100644 --- a/tests/python/relay/test_pass_auto_quantize.py +++ b/tests/python/relay/test_pass_auto_quantize.py @@ -140,7 +140,7 @@ def verify_partition_fails(mod, params): def verify_partition(mod, params): - with relay.quantize.qconfig(**BASE_CFG): + with relay.quantize.qconfig(**BASE_CFG, paritition_conversions='disabled'): unpartitioned_mod = relay.quantize.quantize(mod, params) assert len(unpartitioned_mod.get_global_vars()) == 1, \ 'unpartitioned module should only have one function'