[Sample] Disable multi-threading for TFX demo notebook (#3170)

* [Sample] Disable multi-threading for TFX demo notebook In TFX 0.21.0 release it's not guaranteed to work well with 6 vCPU. Disable it for now. * Update taxi_pipeline_notebook.ipynb Co-authored-by: Yuan (Bob) Gong <gongyuan94@gmail.com>
kubeflow · Feb 26, 2020 · 9a96117 · 9a96117
1 parent 1d42e3f
commit 9a96117
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/samples/core/parameterized_tfx_oss/taxi_pipeline_notebook.ipynb b/samples/core/parameterized_tfx_oss/taxi_pipeline_notebook.ipynb
@@ -239,7 +239,7 @@
     "      trainer, model_analyzer, model_validator, pusher\n",
     "  ],\n",
     "  enable_cache=True,\n",
-    "  beam_pipeline_args=['--direct_num_workers=%d' % 4],\n",
+    "  beam_pipeline_args=['--direct_num_workers=%d' % 0],\n",
     ")"
    ]
   },
@@ -311,4 +311,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}