From ae30ba72fdf26dd02b34f8e734f8f1d8ee30923b Mon Sep 17 00:00:00 2001 From: Logan Weber Date: Tue, 11 Aug 2020 15:22:55 -0700 Subject: [PATCH] partially address feedback --- tests/python/relay/test_pass_auto_quantize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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'