partially address feedback

apache · Aug 11, 2020 · ae30ba7 · ae30ba7
1 parent 938cdef
commit ae30ba7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'