lint

Signed-off-by: xxchan <xxchan22f@gmail.com>
risingwavelabs · Jan 13, 2025 · 0d8e623 · 0d8e623
1 parent 3d1a47b
commit 0d8e623
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/bench/sink_bench/main.rs b/src/bench/sink_bench/main.rs
@@ -56,7 +56,7 @@ use risingwave_connector::sink::{
     SINK_TYPE_APPEND_ONLY, SINK_TYPE_UPSERT,
 };
 use risingwave_connector::source::datagen::{
-    DatagenProperties, DatagenSplit, DatagenSplitEnumerator, DatagenSplitReader,
+    DatagenProperties, DatagenSplitEnumerator, DatagenSplitReader,
 };
 use risingwave_connector::source::{
     Column, DataType, SourceContext, SourceEnumeratorContext, SplitEnumerator, SplitReader,
@@ -304,7 +304,7 @@ impl MockDatagenSource {
             datagen_split_readers.push(
                 DatagenSplitReader::new(
                     properties.clone(),
-                    vec![DatagenSplit::try_from(splits).unwrap()],
+                    vec![splits],
                     parser_config.clone(),
                     SourceContext::dummy().into(),
                     Some(source_schema.clone()),

diff --git a/src/connector/src/macros.rs b/src/connector/src/macros.rs
@@ -320,7 +320,7 @@ macro_rules! impl_connector_properties {
         $crate::paste! {
             impl ConnectorProperties {
                 pub async fn create_split_enumerator(self, context: $crate::source::base::SourceEnumeratorContextRef) -> $crate::error::ConnectorResult<Box<dyn AnySplitEnumerator>> {
-                    use crate::source::prelude::*;
+                    use $crate::source::prelude::*;
                     let enumerator: Box<dyn AnySplitEnumerator> = match self {
                         $(
                             ConnectorProperties::$variant_name(prop) =>