Cargo fmt everything

Author: Kerollmops

benches/benchmark.rs

@@ -1,6 +1,8 @@
-use hannoy::{distances::Cosine, Database, Writer};
+use hannoy::distances::Cosine;
+use hannoy::{Database, Writer};
 use heed::{Env, EnvOpenOptions, RwTxn};
-use rand::{rngs::StdRng, Rng, SeedableRng};
+use rand::rngs::StdRng;
+use rand::{Rng, SeedableRng};
 use tempfile::tempdir;
 
 static M: usize = 16;

benches/speed.rs

@@ -3,14 +3,14 @@ use std::fs::OpenOptions;
 use std::hint::black_box;
 use std::io::Write;
 
-use hannoy::Reader;
-use hannoy::{distances::Cosine, Database, Writer};
+use hannoy::distances::Cosine;
+use hannoy::{Database, Reader, Writer};
 use heed::{Env, EnvOpenOptions, RwTxn};
 use hnsw_rs;
 use hnsw_rs::hnsw::Hnsw;
 use hnsw_rs::prelude::DistCosine;
-use rand::thread_rng;
-use rand::{rngs::StdRng, Rng, SeedableRng};
+use rand::rngs::StdRng;
+use rand::{thread_rng, Rng, SeedableRng};
 use tempfile::tempdir;
 
 static M: usize = 16;

src/distance/hamming.rs

@@ -1,9 +1,10 @@
 use std::fmt;
 
+use bytemuck::{Pod, Zeroable};
+
 use crate::distance::Distance;
 use crate::node::Item;
 use crate::unaligned_vector::{Binary, UnalignedVector};
-use bytemuck::{Pod, Zeroable};
 
 /// The Hamming distance between two vectors is the number of positions at
 /// which the corresponding symbols are different.

src/error.rs

@@ -1,6 +1,9 @@
 use std::io;
 
-use crate::{key::Key, node_id::NodeMode, version::Version, ItemId, LayerId};
+use crate::key::Key;
+use crate::node_id::NodeMode;
+use crate::version::Version;
+use crate::{ItemId, LayerId};
 
 /// The different set of errors that hannoy can encounter.
 #[derive(Debug, thiserror::Error)]

src/hnsw.rs

@@ -173,7 +173,8 @@ impl<'a, D: Distance, const M: usize, const M0: usize> HnswBuilder<'a, D, M, M0>
 
         level_groups.into_iter().try_for_each(|grp| {
             grp.into_par_iter().try_for_each(|&(item_id, lvl)| {
-                if cancel_index.fetch_add(1, Relaxed).is_multiple_of(CANCELLATION_PROBING) && (self.cancel)()
+                if cancel_index.fetch_add(1, Relaxed).is_multiple_of(CANCELLATION_PROBING)
+                    && (self.cancel)()
                 {
                     Err(Error::BuildCancelled)
                 } else {

src/node.rs

@@ -189,11 +189,15 @@ impl fmt::Display for InvalidNodeDecoding {
 
 #[cfg(test)]
 mod tests {
-    use super::{Item, Links, Node, NodeCodec};
-    use crate::{distance::Cosine, internals::UnalignedVector, Distance};
+    use std::borrow::Cow;
+
     use heed::{BytesDecode, BytesEncode};
     use roaring::RoaringBitmap;
-    use std::borrow::Cow;
+
+    use super::{Item, Links, Node, NodeCodec};
+    use crate::distance::Cosine;
+    use crate::internals::UnalignedVector;
+    use crate::Distance;
 
     #[test]
     fn check_bytes_encode_decode() {

src/python.rs

@@ -1,17 +1,15 @@
 //! Python bindings for hannoy.
+use std::path::PathBuf;
+use std::sync::LazyLock;
+
 use heed::{RoTxn, RwTxn, WithoutTls};
 use once_cell::sync::OnceCell;
 use parking_lot::{MappedMutexGuard, Mutex, MutexGuard};
-use pyo3::{
-    exceptions::{PyIOError, PyRuntimeError},
-    prelude::*,
-    types::PyType,
-};
-use pyo3_stub_gen::{
-    define_stub_info_gatherer,
-    derive::{gen_stub_pyclass, gen_stub_pyclass_enum, gen_stub_pymethods},
-};
-use std::{path::PathBuf, sync::LazyLock};
+use pyo3::exceptions::{PyIOError, PyRuntimeError};
+use pyo3::prelude::*;
+use pyo3::types::PyType;
+use pyo3_stub_gen::define_stub_info_gatherer;
+use pyo3_stub_gen::derive::{gen_stub_pyclass, gen_stub_pyclass_enum, gen_stub_pymethods};
 
 use crate::{distance, Database, ItemId, Reader, Writer};
 static DEFAULT_ENV_SIZE: usize = 1024 * 1024 * 1024; // 1GiB
@@ -257,7 +255,8 @@ pub(super) struct PyWriter {
 
 impl PyWriter {
     fn build(&self) -> PyResult<()> {
-        use rand::{rngs::StdRng, SeedableRng};
+        use rand::rngs::StdRng;
+        use rand::SeedableRng;
 
         let mut rng = StdRng::seed_from_u64(42);
         let mut wtxn = get_rw_txn()?;

src/reader.rs

@@ -410,14 +410,15 @@ impl<D: Distance> Reader<D> {
         index: u16,
         metadata: &Metadata,
     ) -> Result<()> {
-        use crate::unaligned_vector::UnalignedVectorCodec;
+        use std::collections::VecDeque;
+        use std::sync::atomic::{AtomicUsize, Ordering};
 
         use heed::types::Bytes;
         use madvise::AccessPattern;
-        use std::collections::VecDeque;
-        use std::sync::atomic::{AtomicUsize, Ordering};
         use tracing::warn;
 
+        use crate::unaligned_vector::UnalignedVectorCodec;
+
         let page_size = page_size::get();
         let mut available_memory: usize = std::env::var(READER_AVAILABLE_MEMORY)
             .ok()

src/spaces/simple_neon.rs

@@ -1,8 +1,9 @@
-#[cfg(target_feature = "neon")]
-use crate::unaligned_vector::UnalignedVector;
 use std::arch::aarch64::*;
 use std::ptr::read_unaligned;
 
+#[cfg(target_feature = "neon")]
+use crate::unaligned_vector::UnalignedVector;
+
 #[cfg(target_feature = "neon")]
 pub(crate) unsafe fn euclid_similarity_neon(
     v1: &UnalignedVector<f32>,

src/tests/fuzz.rs

@@ -1,14 +1,14 @@
-use crate::{
-    distance::Cosine,
-    tests::{create_database_indices_with_items, DatabaseHandle},
-    Database, Reader, Writer,
-};
 use arbitrary::{Arbitrary, Unstructured};
 use heed::RoTxn;
-use rand::{self, rngs::StdRng, Rng, SeedableRng};
+use rand::rngs::StdRng;
+use rand::{self, Rng, SeedableRng};
 use roaring::RoaringBitmap;
 use tracing::info;
 
+use crate::distance::Cosine;
+use crate::tests::{create_database_indices_with_items, DatabaseHandle};
+use crate::{Database, Reader, Writer};
+
 #[derive(Debug)]
 struct Item<const M: usize> {
     id: u32,