style: format to the black default line length

Author: cpcloud

ci/make_geography_db.py

@@ -70,10 +70,7 @@ def make_geography_db(
             table = sa.Table(
                 table_name,
                 metadata,
-                *(
-                    sa.Column(col_name, col_type)
-                    for col_name, col_type in schema
-                ),
+                *(sa.Column(col_name, col_type) for col_name, col_type in schema),
             )
             table_columns = table.c.keys()
             post_parse = POST_PARSE_FUNCTIONS.get(table_name, toolz.identity)
@@ -82,16 +79,11 @@ def make_geography_db(
             table.create(bind=bind)
             bind.execute(
                 table.insert().values(),
-                [
-                    post_parse(dict(zip(table_columns, row)))
-                    for row in data[table_name]
-                ],
+                [post_parse(dict(zip(table_columns, row))) for row in data[table_name]],
             )
 
 
-@click.command(
-    help="Create the geography SQLite database for the Ibis tutorial"
-)
+@click.command(help="Create the geography SQLite database for the Ibis tutorial")
 @click.option(
     "-d",
     "--output-directory",

dev/poetry2setup.py

@@ -35,9 +35,7 @@ def main(args: argparse.Namespace) -> None:
 
 
 if __name__ == "__main__":
-    p = argparse.ArgumentParser(
-        description="Generate a setup.py file from pyproject.toml"
-    )
+    p = argparse.ArgumentParser(description="Generate a setup.py file from pyproject.toml")
     p.add_argument(
         "-i",
         "--input-directory",

docs/example.py

@@ -11,9 +11,7 @@
     .aggregate(
         num_investments=c.permalink.nunique(),
         acq_ipos=(
-            c.status.isin(("ipo", "acquired"))
-            .ifelse(c.permalink, ibis.NA)
-            .nunique()
+            c.status.isin(("ipo", "acquired")).ifelse(c.permalink, ibis.NA).nunique()
         ),
     )
     .mutate(acq_rate=lambda t: t.acq_ipos / t.num_investments)

docs/sqlalchemy_example.py

@@ -12,17 +12,11 @@
             ).label("investor_name"),
             sa.func.count(c.c.permalink.distinct()).label("num_investments"),
             sa.func.count(
-                sa.case(
-                    [(c.status.in_(("ipo", "acquired")), c.c.permalink)]
-                ).distinct()
+                sa.case([(c.status.in_(("ipo", "acquired")), c.c.permalink)]).distinct()
             ).label("acq_ipos"),
         ]
     )
-    .select_from(
-        c.join(
-            i, onclause=c.c.permalink == i.c.company_permalink, isouter=True
-        )
-    )
+    .select_from(c.join(i, onclause=c.c.permalink == i.c.company_permalink, isouter=True))
     .group_by(1)
     .order_by(sa.desc(2))
 )

gen_matrix.py

@@ -11,10 +11,7 @@ def get_backends():
     pyproject = tomli.loads(Path("pyproject.toml").read_text())
     backends = pyproject["tool"]["poetry"]["plugins"]["ibis.backends"]
     del backends["spark"]
-    return [
-        (backend, getattr(ibis, backend))
-        for backend in sorted(backends.keys())
-    ]
+    return [(backend, getattr(ibis, backend)) for backend in sorted(backends.keys())]
 
 
 def get_leaf_classes(op):
@@ -45,9 +42,7 @@ def get_leaf_classes(op):
 
 
 def main():
-    possible_ops = (
-        frozenset(get_leaf_classes(ops.Value)) | INCLUDED_OPS
-    ) - EXCLUDED_OPS
+    possible_ops = (frozenset(get_leaf_classes(ops.Value)) | INCLUDED_OPS) - EXCLUDED_OPS
 
     support = {"operation": [f"`{op.__name__}`" for op in possible_ops]}
     support.update(

ibis/__init__.py

@@ -43,9 +43,7 @@ def __getattr__(name: str) -> BaseBackend:
     the `ibis.backends` entrypoints. If successful, the `ibis.sqlite`
     attribute is "cached", so this function is only called the first time.
     """
-    entry_points = {
-        ep for ep in util.backend_entry_points() if ep.name == name
-    }
+    entry_points = {ep for ep in util.backend_entry_points() if ep.name == name}
 
     if not entry_points:
         msg = f"module 'ibis' has no attribute '{name}'. "

ibis/backends/base/__init__.py

@@ -310,9 +310,7 @@ def database(self, name: str | None = None) -> Database:
         Database
             A database object for the specified database.
         """
-        return self.database_class(
-            name=name or self.current_database, client=self
-        )
+        return self.database_class(name=name or self.current_database, client=self)
 
     @property
     @abc.abstractmethod
@@ -441,9 +439,7 @@ def register_options(cls) -> None:
             try:
                 setattr(options, backend_name, backend_options)
             except ValueError as e:
-                raise exc.BackendConfigurationNotRegistered(
-                    backend_name
-                ) from e
+                raise exc.BackendConfigurationNotRegistered(backend_name) from e
 
     def compile(
         self,
@@ -472,14 +468,10 @@ def add_operation(self, operation: ops.Node) -> Callable:
         ...     return 'NULL'
         """
         if not hasattr(self, 'compiler'):
-            raise RuntimeError(
-                'Only SQL-based backends support `add_operation`'
-            )
+            raise RuntimeError('Only SQL-based backends support `add_operation`')
 
         def decorator(translation_function: Callable) -> None:
-            self.compiler.translator_class.add_operation(
-                operation, translation_function
-            )
+            self.compiler.translator_class.add_operation(operation, translation_function)
 
         return decorator
 
@@ -546,9 +538,7 @@ def drop_table(
         force
             If `False`, an exception is raised if the table does not exist.
         """
-        raise NotImplementedError(
-            f'Backend "{self.name}" does not implement "drop_table"'
-        )
+        raise NotImplementedError(f'Backend "{self.name}" does not implement "drop_table"')
 
     def create_view(
         self,
@@ -569,9 +559,7 @@ def create_view(
             Name of the database where the view will be created, if not the
             default.
         """
-        raise NotImplementedError(
-            f'Backend "{self.name}" does not implement "create_view"'
-        )
+        raise NotImplementedError(f'Backend "{self.name}" does not implement "create_view"')
 
     def drop_view(
         self, name: str, database: str | None = None, force: bool = False
@@ -587,9 +575,7 @@ def drop_view(
         force
             If `False`, an exception is raised if the view does not exist.
         """
-        raise NotImplementedError(
-            f'Backend "{self.name}" does not implement "drop_view"'
-        )
+        raise NotImplementedError(f'Backend "{self.name}" does not implement "drop_view"')
 
     @classmethod
     def has_operation(cls, operation: type[ops.Value]) -> bool:

ibis/backends/base/sql/__init__.py

@@ -103,9 +103,7 @@ def sql(self, query: str) -> ir.Table:
         return ops.SQLQueryResult(query, schema, self).to_expr()
 
     def _get_schema_using_query(self, query):
-        raise NotImplementedError(
-            f"Backend {self.name} does not support .sql()"
-        )
+        raise NotImplementedError(f"Backend {self.name} does not support .sql()")
 
     def raw_sql(self, query: str) -> Any:
         """Execute a query string.
@@ -175,9 +173,7 @@ def execute(
         # feature than all this magic.
         # we don't want to pass `timecontext` to `raw_sql`
         kwargs.pop('timecontext', None)
-        query_ast = self.compiler.to_ast_ensure_limit(
-            expr, limit, params=params
-        )
+        query_ast = self.compiler.to_ast_ensure_limit(expr, limit, params=params)
         sql = query_ast.compile()
         self._log(sql)
 
@@ -236,8 +232,7 @@ def ast_schema(self, query_ast, **kwargs) -> sch.Schema:
             return sch.schema({op.name: op.output_dtype})
         else:
             raise ValueError(
-                'Expression with type {} does not have a '
-                'schema'.format(type(self.expr))
+                'Expression with type {} does not have a ' 'schema'.format(type(self.expr))
             )
 
     def _log(self, sql: str) -> None:
@@ -275,9 +270,7 @@ def compile(
             The output of compilation. The type of this value depends on the
             backend.
         """
-        return self.compiler.to_ast_ensure_limit(
-            expr, limit, params=params
-        ).compile()
+        return self.compiler.to_ast_ensure_limit(expr, limit, params=params).compile()
 
     def explain(
         self,
@@ -323,6 +316,5 @@ def has_operation(cls, operation: type[ops.Value]) -> bool:
 
     def _create_temp_view(self, view, definition):
         raise NotImplementedError(
-            f"The {self.name} backend does not implement temporary view "
-            "creation"
+            f"The {self.name} backend does not implement temporary view " "creation"
         )

ibis/backends/base/sql/alchemy/__init__.py

@@ -15,10 +15,7 @@
 import ibis.expr.types as ir
 import ibis.util as util
 from ibis.backends.base.sql import BaseSQLBackend
-from ibis.backends.base.sql.alchemy.database import (
-    AlchemyDatabase,
-    AlchemyTable,
-)
+from ibis.backends.base.sql.alchemy.database import AlchemyDatabase, AlchemyTable
 from ibis.backends.base.sql.alchemy.datatypes import (
     schema_from_table,
     table_from_schema,
@@ -36,10 +33,7 @@
     varargs,
     variance_reduction,
 )
-from ibis.backends.base.sql.alchemy.translator import (
-    AlchemyContext,
-    AlchemyExprTranslator,
-)
+from ibis.backends.base.sql.alchemy.translator import AlchemyContext, AlchemyExprTranslator
 
 __all__ = (
     'BaseAlchemyBackend',
@@ -71,9 +65,7 @@ class BaseAlchemyBackend(BaseSQLBackend):
     table_class = AlchemyTable
     compiler = AlchemyCompiler
 
-    def _build_alchemy_url(
-        self, url, host, port, user, password, database, driver
-    ):
+    def _build_alchemy_url(self, url, host, port, user, password, database, driver):
         if url is not None:
             return sa.engine.url.make_url(url)
 
@@ -189,8 +181,7 @@ def create_table(
 
         if database is not None:
             raise NotImplementedError(
-                'Creating tables from a different database is not yet '
-                'implemented'
+                'Creating tables from a different database is not yet ' 'implemented'
             )
 
         if expr is None and schema is None:
@@ -236,9 +227,7 @@ def _get_insert_method(self, expr):
 
         return methodcaller("from_select", list(expr.columns), compiled)
 
-    def _columns_from_schema(
-        self, name: str, schema: sch.Schema
-    ) -> list[sa.Column]:
+    def _columns_from_schema(self, name: str, schema: sch.Schema) -> list[sa.Column]:
         return [
             sa.Column(colname, to_sqla_type(dtype), nullable=dtype.nullable)
             for colname, dtype in zip(schema.names, schema.types)
@@ -273,8 +262,7 @@ def drop_table(
 
         if database is not None:
             raise NotImplementedError(
-                'Dropping tables from a different database is not yet '
-                'implemented'
+                'Dropping tables from a different database is not yet ' 'implemented'
             )
 
         t = self._get_sqla_table(table_name, schema=database, autoload=False)
@@ -515,8 +503,7 @@ def _get_temp_view_definition(
         definition: sa.sql.compiler.Compiled,
     ) -> str:
         raise NotImplementedError(
-            f"The {self.name} backend does not implement temporary view "
-            "creation"
+            f"The {self.name} backend does not implement temporary view " "creation"
         )
 
     def _register_temp_view_cleanup(self, name: str, raw_name: str) -> None:

ibis/backends/base/sql/alchemy/database.py

@@ -23,9 +23,7 @@ def __init__(self, source, sqla_table, name, schema):
             name = sqla_table.name
         if schema is None:
             schema = sch.infer(sqla_table, schema=schema)
-        super().__init__(
-            name=name, schema=schema, sqla_table=sqla_table, source=source
-        )
+        super().__init__(name=name, schema=schema, sqla_table=sqla_table, source=source)
 
     # TODO(kszucs): remove this
     def __equals__(self, other: AlchemyTable) -> bool: