[FIX] title font, add ylabel in plotPredictions

hollydinkel · Aug 14, 2024 · b742421 · b742421
1 parent 66f7424
commit b742421
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 2 deletions.
diff --git a/images/gom_isp.png b/images/gom_isp.png
diff --git a/images/iqps_isp.png b/images/iqps_isp.png
diff --git a/images/kleos_isp.png b/images/kleos_isp.png
diff --git a/images/planet_isp.png b/images/planet_isp.png
diff --git a/images/satellogic_isp.png b/images/satellogic_isp.png
diff --git a/src/plot_utils.py b/src/plot_utils.py
@@ -29,10 +29,10 @@ def plotKPIs(fig, ax, data, key, companyMetadata, company, ylabel):
     return fig
 
 def plotPredictions(fig, ax, time, prediction, companyMetadata, company):
-    plt.title(company, fontsize=16)
+    ax.set_title(company, fontsize=16, fontweight='bold')
     ax.tick_params(axis='both', which='major', labelsize=16)
     ax.set_xlabel('Year', fontsize=16)
-    ax.set_ylabel('Revenue', fontsize=16)
+    ax.set_ylabel('Revenue ($ USD)', fontsize=16)
     ax.xaxis.set_major_locator(ticker.MaxNLocator(nbins=5))
     ax.yaxis.set_major_locator(ticker.MaxNLocator(nbins=5))
     company_color = np.array(companyMetadata[company]["color"]) / 255