fix clippy

atoma-network · Apr 10, 2024 · 1fa6475 · 1fa6475
1 parent 0d5c8d8
commit 1fa6475
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/atoma-inference/src/models/candle/llama.rs b/atoma-inference/src/models/candle/llama.rs
@@ -203,7 +203,7 @@ mod tests {
     #[test]
     fn test_llama_model_interface() {
         let api_key = "".to_string();
-        let cache_dir: PathBuf = "./test_llama_cache_dir/".try_into().unwrap();
+        let cache_dir: PathBuf = "./test_llama_cache_dir/".into();
         let model_id = "llama_tiny_llama_1_1b_chat".to_string();
         let dtype = "f32".to_string();
         let revision = "main".to_string();
@@ -249,7 +249,7 @@ mod tests {
             panic!("Invalid device")
         }
 
-        assert_eq!(model.cache.use_kv_cache, true);
+        assert!(model.cache.use_kv_cache);
         assert_eq!(model.model_type, ModelType::LlamaTinyLlama1_1BChat);
 
         let prompt = "Write a hello world rust program: ".to_string();
@@ -275,7 +275,7 @@ mod tests {
         println!("output = {output}");
 
         assert!(output.len() > 1);
-        assert!(output.split(" ").collect::<Vec<_>>().len() <= max_tokens);
+        assert!(output.split(' ').collect::<Vec<_>>().len() <= max_tokens);
 
         std::fs::remove_dir_all(cache_dir).unwrap();
     }

diff --git a/atoma-inference/src/models/candle/mamba.rs b/atoma-inference/src/models/candle/mamba.rs
@@ -233,7 +233,7 @@ mod tests {
     #[test]
     fn test_mamba_model_interface() {
         let api_key = "".to_string();
-        let cache_dir: PathBuf = "./test_mamba_cache_dir/".try_into().unwrap();
+        let cache_dir: PathBuf = "./test_mamba_cache_dir/".into();
         let model_id = "mamba_130m".to_string();
         let dtype = "f32".to_string();
         let revision = "refs/pr/1".to_string();

diff --git a/atoma-inference/src/models/candle/stable_diffusion.rs b/atoma-inference/src/models/candle/stable_diffusion.rs
@@ -625,7 +625,7 @@ mod tests {
     #[tokio::test]
     async fn test_stable_diffusion_model_interface() {
         let api_key = "".to_string();
-        let cache_dir: PathBuf = "./test_sd_cache_dir/".try_into().unwrap();
+        let cache_dir: PathBuf = "./test_sd_cache_dir/".into();
         let model_id = "stable_diffusion_v1-5".to_string();
         let dtype = "f32".to_string();
         let revision = "".to_string();