[beam] branch master updated: [BEAM-12126] Fix DirectRunner not respecting use_deprecated_reads

[boyuanz]

This is an automated email from the ASF dual-hosted git repository.

boyuanz pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new 2dcb7da  [BEAM-12126] Fix DirectRunner not respecting 
use_deprecated_reads
     new 58bd73c  Merge pull request #14469 from [BEAM-12126] Fix DirectRunner 
not respecting use_deprecated_reads
2dcb7da is described below

commit 2dcb7da3add3db01d0fbd14b9b64ea8636eda325
Author: Steve Niemitz <steveniem...@gmail.com>
AuthorDate: Thu Apr 8 00:16:52 2021 -0400

    [BEAM-12126] Fix DirectRunner not respecting use_deprecated_reads
---
 .../src/main/java/org/apache/beam/runners/direct/DirectRunner.java     | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git 
a/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectRunner.java
 
b/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectRunner.java
index 3404fa5..a9a154e 100644
--- 
a/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectRunner.java
+++ 
b/runners/direct-java/src/main/java/org/apache/beam/runners/direct/DirectRunner.java
@@ -184,7 +184,6 @@ public class DirectRunner extends 
PipelineRunner<DirectPipelineResult> {
 
       DisplayDataValidator.validatePipeline(pipeline);
       DisplayDataValidator.validateOptions(options);
-      
SplittableParDo.convertReadBasedSplittableDoFnsToPrimitiveReadsIfNecessary(pipeline);
 
       ExecutorService metricsPool =
           Executors.newCachedThreadPool(
@@ -253,6 +252,8 @@ public class DirectRunner extends 
PipelineRunner<DirectPipelineResult> {
 
     // The last set of overrides includes GBK overrides used in WriteView
     pipeline.replaceAll(groupByKeyOverrides());
+
+    
SplittableParDo.convertReadBasedSplittableDoFnsToPrimitiveReadsIfNecessary(pipeline);
   }
 
   @SuppressWarnings("rawtypes")