Adds kwargs so user can customize vector field estimates (#1036)

apasarkar · web-flow · commit b7c15decdd24 · 2026-04-16T15:49:15.000-04:00
* Adds kwargs so user can customize vector field estimates

* Does the kwarg organization the way ndpositions does

* fixes graphic constructor
diff --git a/fastplotlib/widgets/nd_widget/_nd_vectors.py b/fastplotlib/widgets/nd_widget/_nd_vectors.py
@@ -180,6 +180,7 @@ def __init__(
         spatial_func: Callable[[ArrayProtocol], ArrayProtocol] = None,
         slider_dim_transforms=None,
         name: str = None,
+        graphic_kwargs: dict = None,
     ):
         """
         ``NDGraphic`` subclass for n-dimensional vector rendering
@@ -257,6 +258,11 @@ def __init__(
 
         self._graphic: VectorsGraphic | None = None
 
+        if graphic_kwargs is None:
+            self._graphic_kwargs = dict()
+        else:
+            self._graphic_kwargs = graphic_kwargs
+
         # create a graphic
         self._create_graphic()
 
@@ -293,8 +299,10 @@ def _create_graphic(self):
             self._subplot.delete_graphic(old_graphic)
 
         # create the new graphic
-        self._graphic = self._subplot.add_vectors(
-            positions=data_slice[:, 0], directions=data_slice[:, 1]
+        self._graphic = VectorsGraphic(
+            positions=data_slice[:, 0],
+            directions=data_slice[:, 1],
+            **self._graphic_kwargs
         )
 
         self._subplot.add_graphic(self._graphic)
diff --git a/fastplotlib/widgets/nd_widget/_ndw_subplot.py b/fastplotlib/widgets/nd_widget/_ndw_subplot.py
@@ -94,6 +94,7 @@ def add_nd_vectors(
         spatial_func: Callable[[ArrayProtocol], ArrayProtocol] = None,
         slider_dim_transforms=None,
         name: str = None,
+        **kwargs
     ) -> NDVectors:
         nd = NDVectors(
             self.ndw.indices,
@@ -106,6 +107,7 @@ def add_nd_vectors(
             spatial_func=spatial_func,
             slider_dim_transforms=slider_dim_transforms,
             name=name,
+            **kwargs
         )
 
         self._nd_graphics.append(nd)
