fixed #12 since column was not being used

alvarobartt · Jun 28, 2020 · 7447b75 · 7447b75
1 parent 8692ea9
commit 7447b75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trendet/identification.py b/trendet/identification.py
@@ -523,12 +523,12 @@ def identify_df_trends(df, column, window_size=5, identify='both'):
 
     up_trend = {
         'name': 'Up Trend',
-        'element': np.negative(df['Close'])
+        'element': np.negative(df[column])
     }
 
     down_trend = {
         'name': 'Down Trend',
-        'element': df['Close']
+        'element': df[column]
     }
 
     if identify == 'both':