[vis] Fix type hint for plot()

tum-pbs · Apr 20, 2023 · 759dea5 · 759dea5
1 parent 36cca9f
commit 759dea5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phi/vis/_vis.py b/phi/vis/_vis.py
@@ -266,7 +266,7 @@ def get_current_figure():
     return LAST_FIGURE[0]
 
 
-def plot(*fields: Union[SampledField, Tensor],
+def plot(*fields: Union[SampledField, Tensor, Geometry],
          lib: Union[str, PlottingLibrary] = None,
          row_dims: DimFilter = None,
          col_dims: DimFilter = batch,