Reformat using Black

Author: BenSchZA

model/parts/fei_capital_allocation.py

@@ -84,7 +84,8 @@ def policy_fei_capital_allocation_endogenous_weight_update(
     }
     yield_history: List[List] = yield_history_map.values()
     yield_map = {
-        key: max(sum(yield_history) / moving_average_window, 1e-18) or previous_state[key].yield_rate
+        key: max(sum(yield_history) / moving_average_window, 1e-18)
+        or previous_state[key].yield_rate
         for key, yield_history in yield_history_map.items()
     }
     yield_vector = np.array(list(yield_map.values()))

model/parts/liquidity_pools.py

@@ -99,7 +99,8 @@ def policy_constant_function_market_maker(
 
     # Update User Deposit LP balance
     volatile_liquidity_pool_user_deposit.set_balance(
-        updated_volatile_asset_balance * (1 - protocol_liquidity_share), volatile_asset_price
+        updated_volatile_asset_balance * (1 - protocol_liquidity_share),
+        volatile_asset_price,
     )
     fei_liquidity_pool_user_deposit.set_balance(
         updated_fei_balance * (1 - protocol_liquidity_share), fei_price

model/parts/peg_stability_module.py

@@ -69,7 +69,8 @@ def policy_peg_stability_module(params: Parameters, substep, state_history, prev
                     abs(mint_redeem_pcv_asset_balance) < active_psm_pcv_deposit.balance
                 ), "Insufficient PCV for redemption"
                 active_psm_pcv_deposit.withdraw(
-                    amount=abs(mint_redeem_pcv_asset_balance), asset_price=pcv_asset_price
+                    amount=abs(mint_redeem_pcv_asset_balance),
+                    asset_price=pcv_asset_price,
                 )
                 # Collect PSM redeem fees
                 psm_mint_redeem_fees = psm_redeem_fee * abs(fei_minted_redeemed)

model/parts/system_metrics.py

@@ -31,12 +31,12 @@ def policy_system_metrics(params, substep, state_history, previous_state):
 
     # NOTE: QUIZ 1
     # reserve_ratio = (total_pcv + total_protocol_owned_fei) / total_fei_supply
-    
+
     return {
         "stable_backing_ratio": stable_backing_ratio,
         "stable_pcv_ratio": stable_pcv_ratio,
         "collateralization_ratio": collateralization_ratio,
-        #"reserve_ratio": reserve_ratio, # NOTE: QUIZ 1
+        # "reserve_ratio": reserve_ratio, # NOTE: QUIZ 1
         "pcv_yield_rate": pcv_yield_rate,
         "protocol_equity": protocol_equity,
         "protocol_revenue": protocol_revenue,

model/state_update_blocks.py

@@ -81,7 +81,7 @@
                 "stable_pcv_ratio",
                 "collateralization_ratio",
                 # NOTE: Uncomment Below as part of working exercise in Quiz Notebook 1
-                #"reserve_ratio", 
+                # "reserve_ratio",
                 "pcv_yield_rate",
                 # Protocol System Metrics
                 "protocol_equity",

model/state_variables.py

@@ -173,8 +173,8 @@ class StateVariables:
     collateralization_ratio: Percentage = Uninitialized
     """The percentage collateralization of the total FEI supply"""
     # NOTE: Uncomment Below as part of working exercise in Quiz Notebook 1
-    #reserve_ratio: Percentage = Uninitialized
-    #"""The reserve ratio of the Fei Protocol"""
+    # reserve_ratio: Percentage = Uninitialized
+    # """The reserve ratio of the Fei Protocol"""
     pcv_yield: USD = Uninitialized
     """The per-timestep PCV yield accrued"""
     pcv_yield_rate: Percentage = Uninitialized

model/types.py

@@ -145,7 +145,8 @@ def transfer(self, to, amount, from_asset_price=None, to_asset_price=None):
 
         self.withdraw(amount, from_asset_price)
         to.deposit(
-            amount * from_asset_price / to_asset_price if to_asset_price else amount, to_asset_price
+            amount * from_asset_price / to_asset_price if to_asset_price else amount,
+            to_asset_price,
         )
 
         return self, to