From 3447d650ef7aea6174331df81aa0aa609e2b1bc6 Mon Sep 17 00:00:00 2001 From: cardosofede Date: Thu, 20 Jul 2023 15:48:59 +0200 Subject: [PATCH] (feat) add base pydantic config --- quants_lab/strategy/directional_strategy_base.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/quants_lab/strategy/directional_strategy_base.py b/quants_lab/strategy/directional_strategy_base.py index b75d049..afe2502 100644 --- a/quants_lab/strategy/directional_strategy_base.py +++ b/quants_lab/strategy/directional_strategy_base.py @@ -2,12 +2,16 @@ import os from datetime import datetime from typing import Optional import pandas as pd +from pydantic import BaseModel from quants_lab.labeling.triple_barrier_method import triple_barrier_method class DirectionalStrategyBase: + def __init__(self, config: BaseModel): + self.config = config + def get_data(self, start: Optional[str] = None, end: Optional[str] = None): df = self.get_raw_data() return self.filter_df_by_time(df, start, end)