OnurKerimoglu commited on
Commit
6d0b28e
·
1 Parent(s): d730ea7

technical_analysis: robust figure saving path

Browse files
Files changed (1) hide show
  1. src/technical_analysis.py +5 -2
src/technical_analysis.py CHANGED
@@ -229,7 +229,10 @@ class TechnicalAnalysis():
229
  colstoplot)
230
  plt.tight_layout()
231
  if savefig:
232
- plt.savefig(os.path.join('plots', f'{self.ticker}.png'))
 
 
 
233
  plt.close()
234
  fig = None
235
  return fig
@@ -321,7 +324,7 @@ class TechnicalAnalysis():
321
 
322
  if __name__ == '__main__':
323
  ticker = 'GOOG'
324
- df, fig = TechnicalAnalysis(ticker, plot_ta=True, savefig=False, debug=False).run()
325
  print(f'columns: {df.columns}')
326
 
327
 
 
229
  colstoplot)
230
  plt.tight_layout()
231
  if savefig:
232
+ rootdir = os.path.dirname(os.path.dirname(__file__))
233
+ fname = os.path.join(rootdir, 'plots', f'{self.ticker}.png')
234
+ plt.savefig(fname)
235
+ self.logger.info(f'Saved figure into: {fname}')
236
  plt.close()
237
  fig = None
238
  return fig
 
324
 
325
  if __name__ == '__main__':
326
  ticker = 'GOOG'
327
+ df, fig = TechnicalAnalysis(ticker, plot_ta=True, savefig=True, debug=False).run()
328
  print(f'columns: {df.columns}')
329
 
330