File tree Expand file tree Collapse file tree 3 files changed +8
-4
lines changed
datafusion/physical-plan/src/repartition Expand file tree Collapse file tree 3 files changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -132,6 +132,10 @@ impl RunOpt {
132
132
parquet_options. binary_as_string = true ;
133
133
}
134
134
135
+ config
136
+ . options_mut ( )
137
+ . optimizer
138
+ . prefer_round_robin_repartition = self . prefer_round_robin ;
135
139
let rt_builder = self . common . runtime_env_builder ( ) ?;
136
140
let ctx = SessionContext :: new_with_config_rt ( config, rt_builder. build_arc ( ) ?) ;
137
141
self . register_hits ( & ctx) . await ?;
Original file line number Diff line number Diff line change @@ -126,6 +126,10 @@ impl RunOpt {
126
126
. config ( )
127
127
. with_collect_statistics ( !self . disable_statistics ) ;
128
128
config. options_mut ( ) . optimizer . prefer_hash_join = self . prefer_hash_join ;
129
+ config
130
+ . options_mut ( )
131
+ . optimizer
132
+ . prefer_round_robin_repartition = self . prefer_round_robin ;
129
133
let rt_builder = self . common . runtime_env_builder ( ) ?;
130
134
let ctx = SessionContext :: new_with_config_rt ( config, rt_builder. build_arc ( ) ?) ;
131
135
Original file line number Diff line number Diff line change @@ -1672,11 +1672,7 @@ mod tests {
1672
1672
)
1673
1673
. unwrap ( )
1674
1674
}
1675
-
1676
- #[ cfg( test) ]
1677
- mod test {
1678
1675
use arrow:: compute:: SortOptions ;
1679
- use arrow:: datatypes:: { DataType , Field , Schema } ;
1680
1676
1681
1677
use crate :: source:: DataSourceExec ;
1682
1678
use crate :: union:: UnionExec ;
You can’t perform that action at this time.
0 commit comments