remove sht
This commit is contained in:
@@ -6,7 +6,7 @@ import os
|
|||||||
|
|
||||||
from loguru import logger
|
from loguru import logger
|
||||||
|
|
||||||
## base classes taken from nerfstudio
|
## NOTE: base classes taken from nerfstudio
|
||||||
# Pretty printing class
|
# Pretty printing class
|
||||||
class PrintableConfig:
|
class PrintableConfig:
|
||||||
"""Printable Config defining str function"""
|
"""Printable Config defining str function"""
|
||||||
@@ -55,29 +55,6 @@ class InstantiateConfig(PrintableConfig):
|
|||||||
logger.info(f"[yellow]config saved to: {filename}[/yellow]")
|
logger.info(f"[yellow]config saved to: {filename}[/yellow]")
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class LiveConfig(InstantiateConfig):
|
|
||||||
key_id: str = None
|
|
||||||
"""alpaca key id"""
|
|
||||||
|
|
||||||
key_secret: str = None
|
|
||||||
"""alpaca secret"""
|
|
||||||
|
|
||||||
paper: bool = True
|
|
||||||
"""is paper trading or not"""
|
|
||||||
|
|
||||||
def __post_init__(self):
|
|
||||||
if self.key_id is None:
|
|
||||||
from dotenv import load_dotenv
|
|
||||||
load_dotenv()
|
|
||||||
|
|
||||||
self.key_id = os.getenv("ALPACA_KEY_ID")
|
|
||||||
self.key_secret = os.getenv("ALPACA_KEY_SECRET")
|
|
||||||
|
|
||||||
assert self.key_id is not None, "alpaca key id required!"
|
|
||||||
assert self.key_secret is not None, "alpaca key secret required!"
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def load_config(filename: str, inp_conf = None) -> InstantiateConfig:
|
def load_config(filename: str, inp_conf = None) -> InstantiateConfig:
|
||||||
"""load and overwrite config from file"""
|
"""load and overwrite config from file"""
|
||||||
|
|||||||
Reference in New Issue
Block a user