data_prep/data_load.py

import pandas as pd
import torch

from data_prep.basic_feat import add_daily_return , indexify_date
from data_prep.features import add_all_indicators

def prepare_data(path : str) -> pd.DataFrame:
  df = pd.read_csv(path)

  df = indexify_date(df)
  df = add_daily_return(df)

  return add_all_indicators(df)


def dataframe_to_tensor(df : pd.DataFrame) -> torch.Tensor:
  x_np = df.to_numpy()
  return torch.from_numpy(x_np)