Make Colormap() constructor permit base mcolors.Colormap

lukelbd · lukelbd · commit 056a2e31a056 · 2020-05-09T22:21:28.000-06:00
diff --git a/proplot/colors.py b/proplot/colors.py
@@ -2290,6 +2290,8 @@ def __setitem__(self, key, item):
             item = ListedColormap(
                 item.colors, item.name, item.N
             )
+        elif isinstance(item, mcolors.Colormap):  # base class
+            pass
         else:
             raise ValueError(
                 f'Invalid colormap {item}. Must be instance of '
diff --git a/proplot/constructor.py b/proplot/constructor.py
@@ -476,6 +476,8 @@ def _parse_modification(key, value):
             cmap = pcolors.ListedColormap(
                 arg.colors, arg.name, arg.N
             )
+        elif isinstance(arg, mcolors.Colormap):  # base class
+            cmap = arg
 
         # Dictionary of hue/sat/luminance values or 2-tuples
         elif isinstance(arg, dict):

Original file line number	Diff line number	Diff line change
`@@ -2290,6 +2290,8 @@ def __setitem__(self, key, item):`
`2290`	`2290`	`item = ListedColormap(`
`2291`	`2291`	`item.colors, item.name, item.N`
`2292`	`2292`	`)`
	`2293`	`+ elif isinstance(item, mcolors.Colormap): # base class`
	`2294`	`+ pass`
`2293`	`2295`	`else:`
`2294`	`2296`	`raise ValueError(`
`2295`	`2297`	`f'Invalid colormap {item}. Must be instance of '`
Original file line number	Diff line number	Diff line change
`@@ -476,6 +476,8 @@ def _parse_modification(key, value):`
`476`	`476`	`cmap = pcolors.ListedColormap(`
`477`	`477`	`arg.colors, arg.name, arg.N`
`478`	`478`	`)`
	`479`	`+ elif isinstance(arg, mcolors.Colormap): # base class`
	`480`	`+ cmap = arg`
`479`	`481`
`480`	`482`	`# Dictionary of hue/sat/luminance values or 2-tuples`
`481`	`483`	`elif isinstance(arg, dict):`