mgbam commited on
Commit
61d6ca7
·
verified ·
1 Parent(s): 6439442

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -43,9 +43,9 @@ data_load_state.text("Loading SPY data...done!")
43
 
44
  # Calculate technical indicators: MACD and RSI
45
  macd_indicator = MACD(close=data['Close'])
46
- # Squeeze the returned values to ensure a 1-dimensional array
47
- data['MACD'] = pd.Series(macd_indicator.macd().squeeze(), index=data.index)
48
- data['MACD_signal'] = pd.Series(macd_indicator.macd_signal().squeeze(), index=data.index)
49
 
50
  rsi_indicator = RSIIndicator(close=data['Close'], window=14)
51
  data['RSI'] = rsi_indicator.rsi()
@@ -195,4 +195,4 @@ st.write(f"**Return:** {return_pct:.2f}%")
195
 
196
  st.markdown("""
197
  *This extended demo is for educational purposes only and does not constitute financial advice. Always test your strategies extensively before trading with real money.*
198
- """)
 
43
 
44
  # Calculate technical indicators: MACD and RSI
45
  macd_indicator = MACD(close=data['Close'])
46
+ # Use flatten() to convert to a 1D array
47
+ data['MACD'] = pd.Series(np.array(macd_indicator.macd()).flatten(), index=data.index)
48
+ data['MACD_signal'] = pd.Series(np.array(macd_indicator.macd_signal()).flatten(), index=data.index)
49
 
50
  rsi_indicator = RSIIndicator(close=data['Close'], window=14)
51
  data['RSI'] = rsi_indicator.rsi()
 
195
 
196
  st.markdown("""
197
  *This extended demo is for educational purposes only and does not constitute financial advice. Always test your strategies extensively before trading with real money.*
198
+ """)