FIX: AxesSubplot not found

#8

Use mpl.axes.SubplotBase know to check for type of ax (to plot on)

Author: janjoswig

core/cnn.py

@@ -1496,7 +1496,7 @@ def query_data(self, mode='train'):
 
     def evaluate(
         self,
-        ax: Optional[Type[mpl.axes._subplots.AxesSubplot]] = None,
+        ax: Optional[Type[mpl.axes.SubplotBase]] = None,
         mode: str='train', clusters: Optional[List[int]]=None,
         original: bool=False, plot: str='dots',
         parts: Optional[Tuple[Optional[int], Optional[int], Optional[int]]]=None,