fix clippy

atoma-network · Apr 24, 2024 · 9a2c7aa · 9a2c7aa
1 parent f4b2bc1
commit 9a2c7aa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/atoma-inference/src/service.rs b/atoma-inference/src/service.rs
@@ -107,6 +107,7 @@ impl ModelService {
 
 #[cfg(test)]
 mod tests {
+    use atoma_types::ModelError;
     use ed25519_consensus::VerificationKey as PublicKey;
     use rand::rngs::OsRng;
     use std::io::Write;
@@ -148,19 +149,19 @@ mod tests {
         type Output = ();
         type LoadData = ();
 
-        fn fetch(_: String, _: PathBuf, _: ModelConfig) -> Result<(), crate::models::ModelError> {
+        fn fetch(_: String, _: PathBuf, _: ModelConfig) -> Result<(), ModelError> {
             Ok(())
         }
 
-        fn load(_: Self::LoadData) -> Result<Self, crate::models::ModelError> {
+        fn load(_: Self::LoadData) -> Result<Self, ModelError> {
             Ok(Self {})
         }
 
         fn model_type(&self) -> crate::models::types::ModelType {
             crate::models::types::ModelType::LlamaV1
         }
 
-        fn run(&mut self, _: Self::Input) -> Result<Self::Output, crate::models::ModelError> {
+        fn run(&mut self, _: Self::Input) -> Result<Self::Output, ModelError> {
             Ok(())
         }
     }

diff --git a/atoma-inference/src/tests/mod.rs b/atoma-inference/src/tests/mod.rs
@@ -1,4 +1,6 @@
-use crate::models::{config::ModelConfig, types::ModelType, ModelError, ModelTrait};
+use crate::models::types::ModelType;
+use crate::models::{config::ModelConfig, ModelTrait};
+use atoma_types::ModelError;
 use ed25519_consensus::SigningKey as PrivateKey;
 use std::{path::PathBuf, time::Duration};