Skip to content

Commit 2470d5b

Browse files
committed
style: apply ruff formatting
1 parent 160c974 commit 2470d5b

9 files changed

Lines changed: 149 additions & 168 deletions

File tree

paradedb/sqlalchemy/diagnostics.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,7 @@ def paradedb_indexes(engine: Engine) -> list[dict[str, Any]]:
2222
def paradedb_index_segments(engine: Engine, index: str) -> list[dict[str, Any]]:
2323
"""Return segment metadata for a BM25 index from ``pdb.index_segments()``."""
2424
with engine.connect() as conn:
25-
return _exec_and_collect(
26-
conn, "SELECT * FROM pdb.index_segments(%s::regclass)", [index]
27-
)
25+
return _exec_and_collect(conn, "SELECT * FROM pdb.index_segments(%s::regclass)", [index])
2826

2927

3028
def paradedb_verify_index(
@@ -92,12 +90,7 @@ def paradedb_verify_all_indexes(
9290
sql_parts = ["SELECT * FROM pdb.verify_all_indexes("]
9391
params: list[Any] = []
9492
if named_params:
95-
sql_parts.append(
96-
", ".join(
97-
f"{name} => %s::{pg_type}"
98-
for name, pg_type, _ in named_params
99-
)
100-
)
93+
sql_parts.append(", ".join(f"{name} => %s::{pg_type}" for name, pg_type, _ in named_params))
10194
params.extend(value for _, _, value in named_params)
10295
sql_parts.append(")")
10396

paradedb/sqlalchemy/indexing.py

Lines changed: 5 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,7 @@ def render(self) -> str:
4343

4444
args_sql = [_render_sql_arg(value) for value in self.positional_args]
4545
if self.named_args:
46-
rendered_options = ",".join(
47-
f"{key}={_render_config_value(value)}" for key, value in self.named_args
48-
)
46+
rendered_options = ",".join(f"{key}={_render_config_value(value)}" for key, value in self.named_args)
4947
args_sql.append(_quote_term(rendered_options))
5048
return f"pdb.{self.name}({','.join(args_sql)})"
5149

@@ -91,8 +89,7 @@ def _build_spec(
9189
) -> TokenizerSpec:
9290
if not _VALID_TOKENIZER_NAME_RE.match(name):
9391
raise InvalidArgumentError(
94-
"tokenizer name must be a bare identifier (letters, digits, underscore); "
95-
"pass arguments via args/named_args"
92+
"tokenizer name must be a bare identifier (letters, digits, underscore); pass arguments via args/named_args"
9693
)
9794
if args is not None and isinstance(args, str | bytes):
9895
raise InvalidArgumentError("tokenizer args must be a sequence, not a string")
@@ -715,19 +712,13 @@ def validate_pushdown(stmt: Any) -> list[str]:
715712

716713
whereclause = getattr(stmt, "whereclause", None)
717714
if whereclause is None:
718-
warnings.append(
719-
"No WHERE clause found; query will perform a full table scan without ParadeDB"
720-
)
715+
warnings.append("No WHERE clause found; query will perform a full table scan without ParadeDB")
721716
elif not _inspect.has_paradedb_predicate(whereclause):
722-
warnings.append(
723-
"No ParadeDB predicate found in WHERE clause; query will not use a BM25 index"
724-
)
717+
warnings.append("No ParadeDB predicate found in WHERE clause; query will not use a BM25 index")
725718

726719
order_by = getattr(stmt, "_order_by_clauses", None) or ()
727720
limit = getattr(stmt, "_limit_clause", None)
728721
if order_by and limit is None:
729-
warnings.append(
730-
"ORDER BY is present without LIMIT; top-N pushdown to ParadeDB requires both"
731-
)
722+
warnings.append("ORDER BY is present without LIMIT; top-N pushdown to ParadeDB requires both")
732723

733724
return warnings

paradedb/sqlalchemy/pdb.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,9 @@ def snippets(
6060
if max_num_chars is not None:
6161
named_args.append(("max_num_chars", max_num_chars))
6262
if limit is not None:
63-
named_args.append(("\"limit\"", limit))
63+
named_args.append(('"limit"', limit))
6464
if offset is not None:
65-
named_args.append(("\"offset\"", offset))
65+
named_args.append(('"offset"', offset))
6666
if sort_by is not None:
6767
named_args.append(("sort_by", sort_by))
6868
return PDBFunctionWithNamedArgs("snippets", [field], named_args)

0 commit comments

Comments
 (0)