blacken

Author: jonmmease

packages/python/plotly/_plotly_utils/basevalidators.py

@@ -89,7 +89,13 @@ def copy_to_readonly_numpy_array_or_list(v, kind=None, force_numeric=False):
 
     # u: unsigned int, i: signed int, f: float
     numeric_kinds = {"u", "i", "f"}
-    kind_default_dtypes = {"u": "uint32", "i": "int32", "f": "float64", "O": "object", "U": "U"}
+    kind_default_dtypes = {
+        "u": "uint32",
+        "i": "int32",
+        "f": "float64",
+        "O": "object",
+        "U": "U",
+    }
 
     # Handle pandas Series and Index objects
     if pd and isinstance(v, (pd.Series, pd.Index)):
@@ -191,7 +197,7 @@ def is_homogeneous_array(v):
             if v_numpy.shape == ():
                 return False
             else:
-                return True # v_numpy.dtype.kind in ["u", "i", "f", "M", "U"]
+                return True  # v_numpy.dtype.kind in ["u", "i", "f", "M", "U"]
     return False
 
 
@@ -1320,7 +1326,11 @@ def validate_coerce(self, v, should_raise=True):
             pass
         elif self.array_ok and is_homogeneous_array(v):
             v = copy_to_readonly_numpy_array_or_list(v)
-            if not isinstance(v, list) and self.numbers_allowed() and v.dtype.kind in ["u", "i", "f"]:
+            if (
+                not isinstance(v, list)
+                and self.numbers_allowed()
+                and v.dtype.kind in ["u", "i", "f"]
+            ):
                 # Numbers are allowed and we have an array of numbers.
                 # All good
                 pass

packages/python/plotly/_plotly_utils/tests/validators/test_dataarray_validator.py

@@ -32,8 +32,7 @@ def test_validator_acceptance_simple(val, validator):
 
 
 @pytest.mark.parametrize(
-    "val",
-    [np.array([2, 3, 4]), np.array([[1, 2, 3], [4, 5, 6]])],
+    "val", [np.array([2, 3, 4]), np.array([[1, 2, 3], [4, 5, 6]])],
 )
 def test_validator_acceptance_homogeneous(val, validator):
     coerce_val = validator.validate_coerce(val)
@@ -44,7 +43,11 @@ def test_validator_acceptance_homogeneous(val, validator):
 # Accept object array as list
 @pytest.mark.parametrize(
     "val",
-    [["A", "B", "C"], np.array(["A", "B", "C"], dtype="object"), pd.Series(["a", "b", "c"])]
+    [
+        ["A", "B", "C"],
+        np.array(["A", "B", "C"], dtype="object"),
+        pd.Series(["a", "b", "c"]),
+    ],
 )
 def test_validator_accept_object_array_as_list(val, validator):
     coerce_val = validator.validate_coerce(val)