nice code

Signed-off-by: kaixuanliu <[email protected]>
kaixuanliu · Sep 27, 2024 · 92d2821 · 92d2821
1 parent 6ccc883
commit 92d2821
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/backends/python/src/lib.rs b/backends/python/src/lib.rs
@@ -28,14 +28,10 @@ impl PythonBackend {
         match model_type {
             ModelType::Classifier => {}
             ModelType::Embedding(pool) => {
-                match pool {
-                    Pool::Splade => {
-                        return Err(BackendError::Start(format!("{pool:?} is not supported")));
-                    },
-                    _ => {
-                        pool_type = pool;
-                    }
+                if pool == Pool::Splade {
+                    return Err(BackendError::Start(format!("{pool:?} is not supported")));
                 }
+                pool_type = pool;
             }
         };
 

diff --git a/backends/python/src/management.rs b/backends/python/src/management.rs
@@ -38,7 +38,7 @@ impl BackendProcess {
             Pool::LastToken => "lasttoken",
             Pool::Splade => {
                 return Err(BackendError::Start(format!("{pool:?} is not supported")));
-            },
+            }
         };
 
         // Process args