fix

apache · Dec 3, 2020 · d561662 · d561662
1 parent 12d6dbf
commit d561662
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python/unittest/test_auto_scheduler_feature.py b/tests/python/unittest/test_auto_scheduler_feature.py
@@ -153,7 +153,9 @@ def test_gpu_feature():
             inp.task.workload_key,
             inp.task.target,
             None,
-            auto_scheduler.HardwareParams(100000, 16, 64, 1 << 30, 1 << 30, 1 << 30, 1 << 30, 1 << 30),
+            auto_scheduler.HardwareParams(
+                100000, 16, 64, 1 << 30, 1 << 30, 1 << 30, 1 << 30, 1 << 30
+            ),
         )
 
         state = dag.infer_bound_from_state(inputs[0].state)