Skip to content

Commit

Permalink
add link
Browse files Browse the repository at this point in the history
  • Loading branch information
jitingxu1 committed May 28, 2024
1 parent afd29f5 commit e116b68
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 5 deletions.
7 changes: 4 additions & 3 deletions docs/support_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ def check_backend(backend, exprs):
for expr in exprs:
con.execute(expr)
return True
except ibis.common.exceptions.IbisError:
except (TypeError, ValueError, AttributeError):
return False
else:
try:
Expand Down Expand Up @@ -103,7 +103,7 @@ def make_support_matrix():
}
special_step = {
"Drop": {"support": [], "not_support": []},
"Cast": backend_specific,
"Cast": {"support": ["type-specific"], "not_support": ["type-specific"]},
"MutateAt": backend_specific,
"Mutate": backend_specific,
}
Expand Down Expand Up @@ -132,10 +132,11 @@ def make_support_matrix():
step = getattr(ml, step_dict["name"])(**step_dict["params"])
metadata = ml.core.Metadata(targets=("target",))
step.fit_table(data, metadata)
output = step.transform_table(data)

all_expr = []
if hasattr(step, "_fit_expr"):
all_expr.extend(step._fit_expr) # noqa: SLF001
output = step.transform_table(data)
all_expr.append(output)

if check_backend(backend, all_expr):
Expand Down
9 changes: 8 additions & 1 deletion docs/support_matrix.qmd
Original file line number Diff line number Diff line change
Expand Up @@ -61,14 +61,21 @@ dict(value=len(ibis.util.backend_entry_points()) - 3, color="green", icon="datab
```{python}
from itables import show
from support_matrix import make_support_matrix
import re
matrix = make_support_matrix()
show(
matrix.replace({True: "✔", False: "🚫"}),
matrix.replace({
True: "✔️",
False: "🚫",
"backend-specific": "<a href='https://ibis-project.org/support_matrix' title='Click for backend-specific support'>🔧</a>",
"type-specific": "<a title='check backend specific type'>🔍</a>",
}),
ordering=False,
paging=False,
buttons=["copy", "excel", "csv"],
)
```
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ dynamic = ["version", "description"]
dask = ["dask[dataframe]"]
doc = [
"clickhouse-connect",
"ibis-framework[bigquery,clickhouse,datafusion,druid,exasol,impala,mssql,mysql,oracle,risingwave,pyspark,trino,duckdb,examples]",
"ibis-framework[bigquery,clickhouse,dask,datafusion,druid,exasol,flink,impala,mssql,mysql,oracle,polars,postgres,pyspark,risingwave,snowflake,sqlite,trino,duckdb,examples]",
"itables",
"jupyter",
"quartodoc",
Expand Down

0 comments on commit e116b68

Please sign in to comment.