Added unused multi_processing test param back to have the same number…

… of tests so that the flaky tests pass
awslabs · Sep 29, 2020 · c4bc821 · c4bc821
1 parent cc3a557
commit c4bc821
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/dataset/test_variable_length.py b/test/dataset/test_variable_length.py
@@ -226,9 +226,9 @@ def test_variable_length_stack_zerosize(
 
 
 @pytest.mark.parametrize(
- "array_type, axis", itertools.product(["np", "mx"], [0, 1])
+ "array_type, multi_processing, axis", itertools.product(["np", "mx"], [True, False], [0, 1])
 )
-def test_pad_arrays_axis(array_type, axis: int):
+def test_pad_arrays_axis(array_type, multi_processing, axis: int):
  arrays = [
  d["target"] if array_type == "np" else mx.nd.array(d["target"])
  for d in list(iter(get_dataset()))