Files
drift/data_loader/collections.py
T

57 lines
1.2 KiB
Python

from .types import Path, FileName, DataSource, DataCollection
from utils.helpers import flatten
def transform_to_data_collection(path: str, file_names: list[str]) -> DataCollection:
return list(zip([path] * len(file_names), file_names))
__daily_etf = ["GLD", "IEF", "QQQ", "SPY", "TLT"]
__5min_crypto = [
"BTC_USD",
"ETC_USD",
"ETH_USD",
"LTC_USD",
"TRX_USD",
"XLM_USD",
"XMR_USD",
]
__daily_glassnode = [
"rhodl_ratio",
# 'cvdd',
"nvt_ratio",
"nvt_signal",
"velocity",
"supply_adjusted_cdd",
"binary_cdd",
"supply_adjusted_dormancy",
"puell_multiple",
"asopr",
"reserve_risk",
"sopr",
"cdd",
"asol",
"msol",
"dormancy",
"liveliness",
"relative_unrealized_profit",
"relative_unrealized_loss",
"nupl",
"sth_nupl",
"lth_nupl",
"ssr",
"bvin",
# 'hash_rate'
]
data_collections = dict(
daily_etf=transform_to_data_collection("data/daily_etf", __daily_etf),
fivemin_crypto=transform_to_data_collection("data/5min_crypto", __5min_crypto),
daily_glassnode=transform_to_data_collection(
"data/daily_glassnode", __daily_glassnode
),
)