Figure.plot/Figure.plot3d: Improve the check of the "style" parameter…

… for "v" or "V" (#3603)
GenericMappingTools · Nov 11, 2024 · 763dc07 · 763dc07
1 parent ff9daae
commit 763dc07
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pygmt/src/plot.py b/pygmt/src/plot.py
@@ -213,7 +213,8 @@ def plot(
     if kind == "empty":  # Add more columns for vectors input
         # Parameters for vector styles
         if (
-            kwargs.get("S") is not None
+            isinstance(kwargs.get("S"), str)
+            and len(kwargs["S"]) >= 1
             and kwargs["S"][0] in "vV"
             and is_nonstr_iter(direction)
         ):

diff --git a/pygmt/src/plot3d.py b/pygmt/src/plot3d.py
@@ -197,7 +197,8 @@ def plot3d(
     if kind == "empty":  # Add more columns for vectors input
         # Parameters for vector styles
         if (
-            kwargs.get("S") is not None
+            isinstance(kwargs.get("S"), str)
+            and len(kwargs["S"]) >= 1
             and kwargs["S"][0] in "vV"
             and is_nonstr_iter(direction)
         ):