fix tests

plotly · Jul 5, 2020 · 3390b34 · 3390b34
1 parent 7f44e87
commit 3390b34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/python/plotly/plotly/express/_core.py b/packages/python/plotly/plotly/express/_core.py
@@ -1351,7 +1351,7 @@ def build_dataframe(args, constructor):
                 )
 
     no_color = False
-    if type(args["color"]) == str and args["color"] == NO_COLOR:
+    if type(args.get("color", None)) == str and args["color"] == NO_COLOR:
         no_color = True
         args["color"] = None
     # now that things have been prepped, we do the systematic rewriting of `args`