format

Author: msabvid

src/aave/agents/mod.rs

@@ -8,13 +8,13 @@ use alloy_primitives::U256;
 pub use borrow_agent::BorrowAgent;
 pub use liquidation_agent::LiquidationAgent;
 use rand::RngCore;
+use serde::{Deserialize, Serialize};
 pub use uniswap_agent::UniswapPriceAgent;
 pub use uniswap_noise_agent::UniswapNoiseAgent;
 use verbs_rs::agent::{AgentSet, AgentVec, SimState, SingletonAgent};
 use verbs_rs::contract::Transaction;
 use verbs_rs::env::{Env, Validator};
 use verbs_rs::DB;
-use serde::{Serialize, Deserialize};
 
 #[derive(SimState)]
 pub struct AgentStates {
@@ -30,4 +30,4 @@ pub struct AgentData {
     pub liquidation_agents: Vec<Vec<UserData>>,
     pub uniswap_price_agent: Vec<(i128, i128)>,
     pub uniswap_noise_agents: Vec<Vec<U256>>,
-}
+}

src/aave/mod.rs

@@ -6,9 +6,9 @@ mod initialisation;
 mod protocol;
 pub mod types;
 
+use agents::AgentData;
 use verbs_rs::env::GasPriorityValidator;
 use verbs_rs::{agent::RecordedAgentSet, sim_runner};
-use agents::AgentData;
 
 use self::initialisation::initialise_sim;
 
@@ -30,11 +30,11 @@ pub fn aave_sim_from_fork(
     println!("Running");
 
     sim_runner::run(&mut env, &mut agent_sets, seed, n_steps);
-    let sim_data = AgentData{
+    let sim_data = AgentData {
         borrow_agents: agent_sets.borrow_agents.take_records(),
         liquidation_agents: agent_sets.liquidation_agents.take_records(),
         uniswap_price_agent: agent_sets.uniswap_price_agent.take_records(),
-        uniswap_noise_agents: agent_sets.uniswap_noise_agents.take_records()
+        uniswap_noise_agents: agent_sets.uniswap_noise_agents.take_records(),
     };
     let sim_data = json!({
         "seed": seed,
@@ -48,11 +48,11 @@ pub fn aave_sim(seed: u64, n_steps: usize, params: types::SimParameters) -> serd
     let (mut env, mut agent_sets, _, _, _) = initialise_sim(params, validator);
 
     sim_runner::run(&mut env, &mut agent_sets, seed, n_steps);
-    let sim_data = AgentData{
+    let sim_data = AgentData {
         borrow_agents: agent_sets.borrow_agents.take_records(),
         liquidation_agents: agent_sets.liquidation_agents.take_records(),
         uniswap_price_agent: agent_sets.uniswap_price_agent.take_records(),
-        uniswap_noise_agents: agent_sets.uniswap_noise_agents.take_records()
+        uniswap_noise_agents: agent_sets.uniswap_noise_agents.take_records(),
     };
     let sim_data = json!({
         "seed": seed,

src/main.rs

@@ -2,7 +2,6 @@ use aave::aave_sim;
 use clap::Parser;
 mod aave;
 
-
 #[derive(Parser, Debug)]
 #[command(about, long_about = None)]
 struct Args {