diff --git a/Core/CoreTraidMath.py b/Core/CoreTraidMath.py index e77ff96..71e5e7b 100644 --- a/Core/CoreTraidMath.py +++ b/Core/CoreTraidMath.py @@ -29,13 +29,6 @@ class CoreMath: def __init__(self, base_df, params={ 'dataType':'ohcl', - 'colName':{ - 'open':'open', - 'close':'close', - 'high':'high', - 'low':'low', - 'date':'date' - }, 'action': None, 'actionOptions':{} } @@ -44,7 +37,7 @@ class CoreMath: self.base_df=base_df.reset_index(drop=True) self.params=params if self.params['dataType']=='ohcl': - self.col=self.base_df[self.params['colName'][self.params['actionOptions']['valueType']]] + self.col=self.base_df[[self.params['actionOptions']['valueType']]] elif self.params['dataType']=='series': self.col=self.base_df self.ans=self.getAns() @@ -121,4 +114,5 @@ class CoreMath: return(ans) - \ No newline at end of file + +