Spaces:
Sleeping
Sleeping
Commit
·
6d0b28e
1
Parent(s):
d730ea7
technical_analysis: robust figure saving path
Browse files
src/technical_analysis.py
CHANGED
|
@@ -229,7 +229,10 @@ class TechnicalAnalysis():
|
|
| 229 |
colstoplot)
|
| 230 |
plt.tight_layout()
|
| 231 |
if savefig:
|
| 232 |
-
|
|
|
|
|
|
|
|
|
|
| 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=
|
| 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 |
|