cargo fmt

[generate-random.git] / derive-macro / src / handle_enum.rs

diff --git a/derive-macro/src/handle_enum.rs b/derive-macro/src/handle_enum.rs
index b2b14c0a5f0083a84af30748e8423c1bc3277073..8165ab5cdc8ae694d1afb8cf2792971adb05eb85 100644 (file)
--- a/derive-macro/src/handle_enum.rs
+++ b/derive-macro/src/handle_enum.rs
@@ -1,19 +1,23 @@
+use super::generate_fields;
 use proc_macro2::{Ident, Literal, TokenStream};
 use quote::quote;
 use syn::{DataEnum, Variant};
-use super::generate_fields;
 
 fn variant_weight(variant: &Variant) -> Literal {
     for attr in variant.attrs.iter() {
         if attr.path.is_ident("weight") {
-            return attr.parse_args::<Literal>().expect("expected literal for `#[weight(...)]`")
+            return attr
+                .parse_args::<Literal>()
+                .expect("expected literal for `#[weight(...)]`");
         }
     }
     Literal::u64_suffixed(1)
 }
 
 pub fn generate(name: &Ident, ty: DataEnum) -> TokenStream {
-    let mut variant_weights = ty.variants.into_iter()
+    let mut variant_weights = ty
+        .variants
+        .into_iter()
         .map(|variant| (variant_weight(&variant), variant));
 
     let mut arms = TokenStream::new();