move hyperdrive_interface into eth_bots

elfpy/bots/checkpoint_bot.py

@@ -10,13 +10,13 @@
 from eth_account.account import Account
 from web3.contract.contract import Contract
 
-from elfpy import eth, hyperdrive_interface
+from elfpy import eth
 from elfpy.bots.environment_config import EnvironmentConfig
 from elfpy.eth.accounts.eth_account import EthAgent
 from elfpy.eth.rpc_interface import set_anvil_account_balance
 from elfpy.eth.transactions import smart_contract_read, smart_contract_transact
-from elfpy.hyperdrive_interface.hyperdrive_interface import get_hyperdrive_config
 from elfpy.utils import logs
+from eth_bots import hyperdrive_interface
 
 # The portion of the checkpoint that the bot will wait before attempting to
 # mint a new checkpoint.
@@ -90,7 +90,7 @@ def main() -> None:
     # Run the checkpoint bot. This bot will attempt to mint a new checkpoint
     # every checkpoint after a waiting period. It will poll very infrequently
     # to reduce the probability of needing to mint a checkpoint.
-    config = get_hyperdrive_config(hyperdrive)
+    config = hyperdrive_interface.get_hyperdrive_config(hyperdrive)
     checkpoint_duration = config["checkpointDuration"]
     while True:
         # Get the latest block time and check to see if a new checkpoint should

eth_bots/core/execute_agent_trades.py

@@ -11,7 +11,7 @@
 from web3 import Web3
 from web3.contract.contract import Contract
 
-from elfpy import eth, hyperdrive_interface, types
+from elfpy import eth, types
 from elfpy.eth.accounts import EthAgent
 from elfpy.eth.errors.types import UnknownBlockError
 from elfpy.eth.transactions import smart_contract_preview_transaction
@@ -20,6 +20,7 @@
 from elfpy.types import Quantity, TokenType
 from elfpy.wallet.wallet import Long, Short
 from elfpy.wallet.wallet_deltas import WalletDeltas
+from eth_bots import hyperdrive_interface
 
 # TODO: Fix these up when we refactor this file
 # pylint: disable=too-many-locals

eth_bots/core/setup_experiment.py

@@ -9,10 +9,11 @@
 from web3 import Web3
 from web3.contract.contract import Contract
 
-from elfpy import eth, hyperdrive_interface
+from elfpy import eth
 from elfpy.bots import DEFAULT_USERNAME, EnvironmentConfig
 from elfpy.eth.accounts import EthAgent
 from elfpy.utils import logs
+from eth_bots import hyperdrive_interface
 from eth_bots.core import crash_report
 from eth_bots.core.get_agent_accounts import get_agent_accounts
 from eth_bots.eth_bots_config import get_eth_bots_config

eth_bots/data/acquire_data.py

@@ -12,8 +12,9 @@
 from web3 import Web3
 from web3.contract.contract import Contract
 
-from elfpy import eth, hyperdrive_interface
+from elfpy import eth
 from elfpy.utils import logs as log_utils
+from eth_bots import hyperdrive_interface
 from eth_bots.data import convert_data, postgres
 
 # pylint: disable=too-many-arguments

eth_bots/data/convert_data.py

@@ -12,7 +12,8 @@
 from web3.contract.contract import Contract
 from web3.types import BlockData
 
-from elfpy import eth, hyperdrive_interface
+from elfpy import eth
+from eth_bots import hyperdrive_interface
 from eth_bots.data import db_schema
 
 # pylint: disable=too-many-arguments

eth_bots/fund_bots.py

@@ -7,7 +7,8 @@
 from dotenv import load_dotenv
 from eth_account.account import Account
 
-from elfpy import eth, hyperdrive_interface
+from elfpy import eth
+from eth_bots import hyperdrive_interface
 from eth_bots.eth_bots_config import get_eth_bots_config
 
 if __name__ == "__main__":

elfpy/hyperdrive_interface/__init__.py → eth_bots/hyperdrive_interface/__init__.py

@@ -1,4 +1,4 @@
-"""Interfaces for elfpy bots and hyperdrive smart contracts"""
+"""Interfaces for bots and hyperdrive smart contracts."""
 
 from .hyperdrive_addresses import HyperdriveAddresses
 from .hyperdrive_assets import AssetIdPrefix, decode_asset_id, encode_asset_id

eth_bots/streamlit/extract_data_logs.py

@@ -8,7 +8,7 @@
 import numpy as np
 import pandas as pd
 
-from elfpy.hyperdrive_interface import AssetIdPrefix
+from eth_bots.hyperdrive_interface import AssetIdPrefix
 
 
 def read_json_to_pd(json_file):