cargo fmt (#571)

montanalow · web-flow · commit 0c0f39e41744 · 2023-03-23T09:40:30.000-07:00
diff --git a/pgml-extension/src/api.rs b/pgml-extension/src/api.rs
@@ -440,7 +440,10 @@ fn predict_i64(project_name: &str, features: Vec<i64>) -> f32 {
 
 #[pg_extern(strict, name = "predict")]
 fn predict_bool(project_name: &str, features: Vec<bool>) -> f32 {
-    predict_f32(project_name, features.iter().map(|&i| i as u8 as f32).collect())
+    predict_f32(
+        project_name,
+        features.iter().map(|&i| i as u8 as f32).collect(),
+    )
 }
 
 #[pg_extern(strict, name = "predict_proba")]
@@ -579,16 +582,8 @@ pub fn transform_string(
 
 #[cfg(feature = "python")]
 #[pg_extern(name = "generate")]
-fn generate(
-    project_name: &str,
-    inputs: &str,
-    config: default!(JsonB, "'{}'"),
-) -> String {
-    generate_batch(
-        project_name,
-        Vec::from([inputs]),
-        config,
-    )
+fn generate(project_name: &str, inputs: &str, config: default!(JsonB, "'{}'")) -> String {
+    generate_batch(project_name, Vec::from([inputs]), config)
         .first()
         .unwrap()
         .to_string()
diff --git a/pgml-extension/src/bindings/transformers.rs b/pgml-extension/src/bindings/transformers.rs
@@ -83,11 +83,7 @@ pub fn tune(
     metrics
 }
 
-pub fn generate(
-    model_id: i64,
-    inputs: Vec<&str>,
-    config: JsonB,
-) -> Vec<String> {
+pub fn generate(model_id: i64, inputs: Vec<&str>, config: JsonB) -> Vec<String> {
     Python::with_gil(|py| -> Vec<String> {
         let generate = PY_MODULE.getattr(py, "generate").unwrap();
         let config = serde_json::to_string(&config.0).unwrap();
