Update import and return type in moved functions

Author: CGodiksen

crates/modelardb_server/src/remote.rs

@@ -228,7 +228,7 @@ pub fn flight_data_to_record_batch(
     flight_data: &FlightData,
     schema: &Arc<Schema>,
     dictionaries_by_id: &HashMap<i64, ArrayRef>,
-) -> std::result::Result<RecordBatch, Status> {
+) -> StdResult<RecordBatch, Status> {
     debug_assert_eq!(flight_data.flight_descriptor, None);
 
     utils::flight_data_to_arrow_batch(flight_data, schema.clone(), dictionaries_by_id)
@@ -239,7 +239,7 @@ pub fn flight_data_to_record_batch(
 /// [`Status`] that specifies that the table name is missing.
 pub fn table_name_from_flight_descriptor(
     flight_descriptor: &FlightDescriptor,
-) -> std::result::Result<&String, Status> {
+) -> StdResult<&String, Status> {
     flight_descriptor
         .path
         .first()

crates/modelardb_server/tests/integration_test.rs

@@ -36,7 +36,7 @@ use datafusion::arrow::ipc::convert;
 use datafusion::arrow::ipc::writer::{DictionaryTracker, IpcDataGenerator, IpcWriteOptions};
 use datafusion::arrow::record_batch::RecordBatch;
 use futures::{StreamExt, stream};
-use modelardb_test::data_generation;
+use modelardb_test::data_generation::generate_multivariate_time_series;
 use modelardb_test::table::{
     NORMAL_TABLE_NAME, TIME_SERIES_TABLE_NAME, table_metadata_protobuf_bytes,
 };
@@ -265,15 +265,14 @@ impl TestContext {
         multiply_noise_range: Option<Range<f32>>,
         maybe_tag: Option<&str>,
     ) -> RecordBatch {
-        let (uncompressed_timestamps, mut uncompressed_values) =
-            data_generation::generate_multivariate_time_series(
-                TIME_SERIES_TEST_LENGTH,
-                5,
-                SEGMENT_TEST_MINIMUM_LENGTH..2 * SEGMENT_TEST_MINIMUM_LENGTH + 1,
-                generate_irregular_timestamps,
-                multiply_noise_range,
-                100.0..200.0,
-            );
+        let (uncompressed_timestamps, mut uncompressed_values) = generate_multivariate_time_series(
+            TIME_SERIES_TEST_LENGTH,
+            5,
+            SEGMENT_TEST_MINIMUM_LENGTH..2 * SEGMENT_TEST_MINIMUM_LENGTH + 1,
+            generate_irregular_timestamps,
+            multiply_noise_range,
+            100.0..200.0,
+        );
 
         let time_series_len = uncompressed_timestamps.len();