wait on create function embedding (#311)

2018-11-06 14:37:11 -08:00 · 2018-11-06 14:37:11 -08:00 · 11879e2ff1
parent df278567f0
commit 11879e2ff1
1 changed files with 1 additions and 1 deletions
--- a/code_search/src/code_search/dataflow/cli/create_function_embeddings.py
+++ b/code_search/src/code_search/dataflow/cli/create_function_embeddings.py
@ -45,7 +45,7 @@ def create_function_embeddings(argv=None):
  )

  result = pipeline.run()
-  if args.runner == 'DirectRunner':
+  if args.wait_until_finish:
    result.wait_until_finish()