fix for running int tests all green (#247)

NVIDIA-Merlin · Aug 26, 2020 · c0356a3 · c0356a3
1 parent ee379e0
commit c0356a3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/integration/test_notebooks.py b/tests/integration/test_notebooks.py
@@ -15,6 +15,7 @@
 def test_criteo_notebook(tmpdir):
     input_path = os.path.join(DATA_START, "criteo/crit_int_pq")
     output_path = os.path.join(DATA_START, "criteo/crit_test")
+    os.environ["NUM_PARTS"] = "1"
 
     _run_notebook(
         tmpdir,