@@ -24,19 +24,18 @@ use std::{any::Any, sync::Arc};
24
24
use super :: helpers:: { expr_applicable_for_cols, pruned_partition_list, split_files} ;
25
25
use super :: PartitionedFile ;
26
26
27
- use crate :: catalog :: TableProvider ;
28
- use crate :: datasource:: { create_ordering, get_statistics_with_limit, TableType } ;
27
+ use super :: ListingTableUrl ;
28
+ use crate :: datasource:: { create_ordering, get_statistics_with_limit} ;
29
29
use crate :: datasource:: {
30
30
file_format:: { file_compression_type:: FileCompressionType , FileFormat } ,
31
- listing:: ListingTableUrl ,
32
31
physical_plan:: { FileScanConfig , FileSinkConfig } ,
33
32
} ;
34
- use crate :: {
35
- error :: { DataFusionError , Result } ,
36
- execution :: context :: SessionState ,
37
- logical_expr :: { utils :: conjunction , Expr , TableProviderFilterPushDown } ,
38
- physical_plan :: { empty :: EmptyExec , ExecutionPlan , Statistics } ,
39
- } ;
33
+ use crate :: execution :: context :: SessionState ;
34
+ use datafusion_catalog :: TableProvider ;
35
+ use datafusion_common :: { DataFusionError , Result } ;
36
+ use datafusion_expr :: TableType ;
37
+ use datafusion_expr :: { utils :: conjunction , Expr , TableProviderFilterPushDown } ;
38
+ use datafusion_physical_plan :: { empty :: EmptyExec , ExecutionPlan , Statistics } ;
40
39
41
40
use arrow:: datatypes:: { DataType , Field , SchemaBuilder , SchemaRef } ;
42
41
use arrow_schema:: Schema ;
@@ -322,7 +321,7 @@ impl ListingOptions {
322
321
/// ```
323
322
///
324
323
/// [Hive Partitioning]: https://docs.cloudera.com/HDPDocuments/HDP2/HDP-2.1.3/bk_system-admin-guide/content/hive_partitioned_tables.html
325
- /// [`wrap_partition_type_in_dict`]: crate::datasource::physical_plan ::wrap_partition_type_in_dict
324
+ /// [`wrap_partition_type_in_dict`]: datafusion_physical_plan ::wrap_partition_type_in_dict
326
325
pub fn with_table_partition_cols (
327
326
mut self ,
328
327
table_partition_cols : Vec < ( String , DataType ) > ,
@@ -1051,12 +1050,12 @@ mod tests {
1051
1050
use crate :: datasource:: file_format:: parquet:: ParquetFormat ;
1052
1051
use crate :: datasource:: { provider_as_source, MemTable } ;
1053
1052
use crate :: execution:: options:: ArrowReadOptions ;
1054
- use crate :: physical_plan:: collect;
1055
1053
use crate :: prelude:: * ;
1056
1054
use crate :: {
1057
1055
assert_batches_eq,
1058
1056
test:: { columns, object_store:: register_test_store} ,
1059
1057
} ;
1058
+ use datafusion_physical_plan:: collect;
1060
1059
1061
1060
use arrow:: record_batch:: RecordBatch ;
1062
1061
use arrow_schema:: SortOptions ;
@@ -1154,10 +1153,8 @@ mod tests {
1154
1153
let options = ListingOptions :: new ( Arc :: new ( ParquetFormat :: default ( ) ) ) ;
1155
1154
let schema = options. infer_schema ( & state, & table_path) . await . unwrap ( ) ;
1156
1155
1157
- use crate :: {
1158
- datasource:: file_format:: parquet:: ParquetFormat ,
1159
- physical_plan:: expressions:: col as physical_col,
1160
- } ;
1156
+ use crate :: datasource:: file_format:: parquet:: ParquetFormat ;
1157
+ use datafusion_physical_plan:: expressions:: col as physical_col;
1161
1158
use std:: ops:: Add ;
1162
1159
1163
1160
// (file_sort_order, expected_result)
0 commit comments