properly error out

huggingface · Nov 14, 2023 · c646978 · c646978
1 parent 88d862b
commit c646978
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/bindings/python/src/pre_tokenizers.rs b/bindings/python/src/pre_tokenizers.rs
@@ -536,10 +536,9 @@ impl PyMetaspace {
 
         // If a prepend scheme is provided, set it
         if let Some(prepend_scheme) = prepend_scheme {
-            if let Ok(prepend_scheme_enum) = _from_string(prepend_scheme) {
-                new_instance.set_prepend_scheme(prepend_scheme_enum);
-            } else {
-                return Err(err);
+            match _from_string(prepend_scheme) {
+                Ok(prepend_scheme_enum) => new_instance.set_prepend_scheme(prepend_scheme_enum),
+                Err(err) => return Err(err),
             }
         }
         Ok((PyMetaspace {}, new_instance.into()))