fix plotting bug

Author: agosztolai

MARBLE/plotting.py

@@ -277,7 +277,7 @@ def embedding(
         ax.set_axis_off()
 
     if cbar_visible and cbar is not None:
-        plt.colorbar(cbar)
+        plt.colorbar(cbar, ax=ax)
 
     return ax
 

examples/toy_examples/ex_vector_field_flat_surface.py

@@ -52,8 +52,8 @@ def main():
     data = preprocessing.construct_dataset(x, y)
 
     # train model
-    model = net(data, params={'inner_product_features': True, 
-                              'diffusion': False,})
+    model = net(data, params={'inner_product_features': False, 
+                              'diffusion': False})
     model.fit(data)
 
     # evaluate model on data