fixes

Author: mihaimarcu2004

examples/system_setup.py

@@ -4,13 +4,13 @@
 import eth_account
 import lighter
 from utils import save_api_key_config
-from lighter import get_endpoint_profile
+from lighter import get_endpoint_profile, MAINNET, ROBINHOOD, TESTNET
 
 logging.basicConfig(level=logging.DEBUG)
 
 # this is a dummy private key registered on Testnet.
 # It serves as a good example
-ENDPOINT_PROFILE = get_endpoint_profile["TESTNET"]
+ENDPOINT_PROFILE = TESTNET
 BASE_URL = "https://testnet.zklighter.elliot.ai"
 ETH_PRIVATE_KEY = "1234567812345678123456781234567812345678123456781234567812345678"
 API_KEY_INDEX = 3

lighter/__init__.py

@@ -170,7 +170,7 @@
 from lighter.models.withdraw_history import WithdrawHistory
 from lighter.models.withdraw_history_item import WithdrawHistoryItem
 from lighter.models.zk_lighter_info import ZkLighterInfo
-from lighter.endpoint_profiles import  (EndpointProfile, MAINNET, ROBINHOOD, DEFAULT_ENDPOINT_PROFILE, ENDPOINT_PROFILES, get_endpoint_profile)
+from lighter.endpoint_profiles import  (EndpointProfile, MAINNET, ROBINHOOD, TESTNET, DEFAULT_ENDPOINT_PROFILE, ENDPOINT_PROFILES, get_endpoint_profile)
 
 # manual additions
 from lighter.ws_client import WsClient