Linting fix

delvtech · Jun 27, 2024 · 5696e63 · 5696e63
1 parent dbd90f1
commit 5696e63
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/agent0/traiderdaive/gym_environments/full_hyperdrive_env.py b/src/agent0/traiderdaive/gym_environments/full_hyperdrive_env.py
@@ -113,10 +113,8 @@ def __init__(
             chain_port = 10002
 
         local_chain_config = LocalChain.Config(
-            block_timestamp_interval=12,
-            db_port=db_port,
-            chain_port=chain_port,
-            calc_pnl=False)
+            block_timestamp_interval=12, db_port=db_port, chain_port=chain_port, calc_pnl=False
+        )
 
         initial_pool_config = LocalHyperdrive.Config()
         self.chain = LocalChain(local_chain_config)
@@ -553,9 +551,8 @@ def _calculate_reward(self) -> float:
         # The total delta for this episode
 
         current_wallet = self.interactive_hyperdrive.get_positions(
-            show_closed_positions=True,
-            calc_pnl=True,
-            coerce_float=True)
+            show_closed_positions=True, calc_pnl=True, coerce_float=True
+        )
         # Filter by rl bot
         rl_bot_wallet = current_wallet[current_wallet["wallet_address"] == self.rl_bot.address]
         # The rl_bot_wallet shows the pnl of all positions