ruff format changes

vega · Oct 27, 2023 · 71e3598 · 71e3598
1 parent 6ff2b2e
commit 71e3598
Show file tree

Hide file tree

Showing 6 changed files with 24 additions and 21 deletions.
diff --git a/altair/_magics.py b/altair/_magics.py
@@ -58,8 +58,9 @@ def _get_variable(name):
         )
     if name not in ip.user_ns:
         raise NameError(
-            "argument '{}' does not match the "
-            "name of any defined variable".format(name)
+            "argument '{}' does not match the " "name of any defined variable".format(
+                name
+            )
         )
     return ip.user_ns[name]
 

diff --git a/altair/utils/core.py b/altair/utils/core.py
@@ -380,18 +380,21 @@ def to_list_if_array(val):
             # geopandas >=0.6.1 uses the dtype geometry. Continue here
             # otherwise it will give an error on np.issubdtype(dtype, np.integer)
             continue
-        elif dtype_name in {
-            "Int8",
-            "Int16",
-            "Int32",
-            "Int64",
-            "UInt8",
-            "UInt16",
-            "UInt32",
-            "UInt64",
-            "Float32",
-            "Float64",
-        }:  # nullable integer datatypes (since 24.0) and nullable float datatypes (since 1.2.0)
+        elif (
+            dtype_name
+            in {
+                "Int8",
+                "Int16",
+                "Int32",
+                "Int64",
+                "UInt8",
+                "UInt16",
+                "UInt32",
+                "UInt64",
+                "Float32",
+                "Float64",
+            }
+        ):  # nullable integer datatypes (since 24.0) and nullable float datatypes (since 1.2.0)
             # https://pandas.pydata.org/pandas-docs/version/0.25/whatsnew/v0.24.0.html#optional-integer-na-support
             col = df[col_name].astype(object)
             df[col_name] = col.where(col.notnull(), None)

diff --git a/altair/utils/schemapi.py b/altair/utils/schemapi.py
@@ -1079,7 +1079,7 @@ def from_json(
         cls,
         json_string: str,
         validate: bool = True,
-        **kwargs: Any
+        **kwargs: Any,
         # Type hints for this method would get rather complicated
         # if we want to provide a more specific return type
     ) -> Any:

diff --git a/sphinxext/utils.py b/sphinxext/utils.py
@@ -122,8 +122,9 @@ def get_docstring_and_rest(filename):
 
     if not isinstance(node, ast.Module):
         raise TypeError(
-            "This function only supports modules. "
-            "You provided {}".format(node.__class__.__name__)
+            "This function only supports modules. " "You provided {}".format(
+                node.__class__.__name__
+            )
         )
     try:
         # In python 3.7 module knows its docstring.

diff --git a/tests/test_magics.py b/tests/test_magics.py
@@ -31,9 +31,7 @@
 import pandas as pd
 table = pd.DataFrame.from_records({})
 the_data = table
-""".format(
-            DATA_RECORDS
-        )
+""".format(DATA_RECORDS)
     )
 
 

diff --git a/tools/schemapi/schemapi.py b/tools/schemapi/schemapi.py
@@ -1077,7 +1077,7 @@ def from_json(
         cls,
         json_string: str,
         validate: bool = True,
-        **kwargs: Any
+        **kwargs: Any,
         # Type hints for this method would get rather complicated
         # if we want to provide a more specific return type
     ) -> Any: