jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-29 05:21:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

more linting

Browse Source
This commit is contained in:
Jamie
2021-04-13 10:51:37 -04:00
parent 535e89f8b4
commit ac27fe2da7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gamestonk_terminal/portfolio_optimization/port_opt_helper.py
View File

@@ -31,7 +31,6 @@ def process_stocks(list_of_stocks: List[str], period: str = "3mo") -> pd.DataFra
"""
stock_prices = yf.download(list_of_stocks, period=period, group_by="ticker")
stock_closes = pd.DataFrame(index=stock_prices.index)
# process df
for stock in list_of_stocks:
stock_closes[stock] = stock_prices[stock]["Adj Close"]
return stock_closes
@@ -81,15 +80,16 @@ def pie_chart_weights(weights: dict):
if not weights:
return
stocks = list(weights.keys())
sizes = list(weights.values())
fig, ax = plt.subplots(figsize=plot_autoscale(), dpi=PLOT_DPI)
_, ax = plt.subplots(figsize=plot_autoscale(), dpi=PLOT_DPI)
if math.isclose(sum(sizes), 1, rel_tol=0.1):
wedges, texts, autotexts = ax.pie(
wedges, _, autotexts = ax.pie(
sizes, labels=stocks, autopct="%1.1f%%", textprops=dict(color="w")
)
else:
wedges, texts, autotexts = ax.pie(
wedges, _, autotexts = ax.pie(
sizes, labels=stocks, autopct="", textprops=dict(color="w")
)
for i, a in enumerate(autotexts):