Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
robert3005 committed Jul 10, 2024
1 parent 26e0166 commit db86bbe
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 11 deletions.
9 changes: 4 additions & 5 deletions bench-vortex/src/reader.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,11 +5,11 @@ use std::path::{Path, PathBuf};
use std::process::Command;
use std::sync::Arc;

use arrow_array::types::Int64Type;
use arrow_array::{
ArrayRef as ArrowArrayRef, PrimitiveArray as ArrowPrimitiveArray, RecordBatch,
RecordBatchReader,
};
use arrow_array::types::Int64Type;
use arrow_select::concat::concat_batches;
use arrow_select::take::take_record_batch;
use bytes::{Bytes, BytesMut};
Expand All @@ -22,21 +22,20 @@ use parquet::arrow::async_reader::{AsyncFileReader, ParquetObjectReader};
use parquet::arrow::ParquetRecordBatchStreamBuilder;
use serde::{Deserialize, Serialize};
use stream::StreamExt;

use vortex::{Array, IntoArray, IntoCanonical, ToArrayData};
use vortex::array::chunked::ChunkedArray;
use vortex::array::primitive::PrimitiveArray;
use vortex::arrow::FromArrowType;
use vortex::compress::CompressionStrategy;
use vortex::stream::ArrayStreamExt;
use vortex::{Array, IntoArray, IntoCanonical, ToArrayData};
use vortex_buffer::Buffer;
use vortex_dtype::DType;
use vortex_error::{vortex_err, VortexResult};
use vortex_ipc::chunked_reader::ChunkedArrayReader;
use vortex_ipc::io::{TokioAdapter, VortexReadAt, VortexWrite};
use vortex_ipc::io::ObjectStoreExt;
use vortex_ipc::MessageReader;
use vortex_ipc::io::{TokioAdapter, VortexReadAt, VortexWrite};
use vortex_ipc::writer::ArrayWriter;
use vortex_ipc::MessageReader;
use vortex_sampling_compressor::SamplingCompressor;

use crate::{COMPRESSORS, CTX};
Expand Down
5 changes: 2 additions & 3 deletions encodings/runend/src/runend.rs
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
use serde::{Deserialize, Serialize};

use vortex::{ArrayDType, Canonical, impl_encoding, IntoArrayVariant, IntoCanonical};
use vortex::array::primitive::{Primitive, PrimitiveArray};
use vortex::compute::search_sorted::{search_sorted, SearchSortedSide};
use vortex::compute::unary::scalar_at::scalar_at;
use vortex::stats::{ArrayStatistics, ArrayStatisticsCompute};
use vortex::validity::{ArrayValidity, LogicalValidity, Validity, ValidityMetadata};
use vortex::visitor::{AcceptArrayVisitor, ArrayVisitor};
use vortex::{impl_encoding, ArrayDType, Canonical, IntoArrayVariant, IntoCanonical};
use vortex_error::vortex_bail;

use crate::compress::{runend_decode, runend_encode};
Expand Down Expand Up @@ -139,10 +138,10 @@ impl ArrayTrait for RunEndArray {

#[cfg(test)]
mod test {
use vortex::{ArrayDType, ArrayTrait, IntoArray, IntoCanonical};
use vortex::compute::slice::slice;
use vortex::compute::unary::scalar_at::scalar_at;
use vortex::validity::Validity;
use vortex::{ArrayDType, ArrayTrait, IntoArray, IntoCanonical};
use vortex_dtype::{DType, Nullability, PType};

use crate::RunEndArray;
Expand Down
5 changes: 2 additions & 3 deletions vortex-ipc/src/message_writer.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,16 +2,15 @@ use std::io;

use flatbuffers::FlatBufferBuilder;
use itertools::Itertools;

use vortex::Array;
use vortex_buffer::Buffer;
use vortex_buffer::io_buf::IoBuf;
use vortex_buffer::Buffer;
use vortex_dtype::DType;
use vortex_flatbuffers::WriteFlatBuffer;

use crate::ALIGNMENT;
use crate::io::VortexWrite;
use crate::messages::{IPCChunk, IPCMessage, IPCPage, IPCSchema};
use crate::ALIGNMENT;

const ZEROS: [u8; 512] = [0u8; 512];

Expand Down

0 comments on commit db86bbe

Please sign in to comment.