Updated the windfarm map to be compatible with recent releases of mat…

…plotlib
AlgTUDelft · Sep 14, 2023 · a6647cf · a6647cf
1 parent f8ffa96
commit a6647cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/wind_farm_gym/wind_map.py b/wind_farm_gym/wind_map.py
@@ -22,8 +22,8 @@ def __init__(self, size, dpi, cut_plane, turbines_raw_data, wind_speed_limits=No
         self.fig.subplots_adjust(top=1, bottom=0, right=1, left=0, hspace=0, wspace=0)
         self.ax = self.fig.gca()
         self.ax.set_axis_off()
-        self.ax.xaxis.set_major_locator(plt.NullLocator())
-        self.ax.yaxis.set_major_locator(plt.NullLocator())
+        self.ax.xaxis.set_visible(False)
+        self.ax.yaxis.set_visible(False)
         self.canvas = FigureCanvasAgg(self.fig)
         self.color_map = color_map
         if wind_speed_limits is None:
@@ -67,7 +67,7 @@ def render(self):
         self.draw()
 
     def clear(self):
-        self.ax.collections.clear()
+        self.ax.clear()
 
     def close(self):
         plt.close(self.fig)