Allow non-string color when input data is a matrix or a file for plot…

… and plot3d
GenericMappingTools · Sep 19, 2021 · 3aaccd9 · 3aaccd9
1 parent f7d5526
commit 3aaccd9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pygmt/src/plot.py b/pygmt/src/plot.py
@@ -239,7 +239,7 @@ def plot(self, x=None, y=None, data=None, size=None, direction=None, **kwargs):
                 kwargs["S"] = "s0.2c"
         except FileNotFoundError:
             pass
-    if "G" in kwargs and not isinstance(kwargs["G"], str):
+    if "G" in kwargs and is_nonstr_iter(kwargs["G"]):
         if kind != "vectors":
             raise GMTInvalidInput(
                 "Can't use arrays for color if data is matrix or file."

diff --git a/pygmt/src/plot3d.py b/pygmt/src/plot3d.py
@@ -209,7 +209,7 @@ def plot3d(
                 kwargs["S"] = "u0.2c"
         except FileNotFoundError:
             pass
-    if "G" in kwargs and not isinstance(kwargs["G"], str):
+    if "G" in kwargs and is_nonstr_iter(kwargs["G"]):
         if kind != "vectors":
             raise GMTInvalidInput(
                 "Can't use arrays for color if data is matrix or file."