Skip to content

Commit

Permalink
pre-commit hooks
Browse files Browse the repository at this point in the history
  • Loading branch information
zilto authored and zilto committed Mar 3, 2024
1 parent 769e327 commit 85d7554
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 11 deletions.
2 changes: 1 addition & 1 deletion examples/ibis/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ def main(level: str, model: str):
from hamilton import driver
from hamilton.execution.executors import SynchronousLocalTaskExecutor
from hamilton.plugins.h_tqdm import ProgressBar

parser = argparse.ArgumentParser()
parser.add_argument("--level", choices=["column", "table"])
parser.add_argument("--model", choices=["linear", "random_forest", "boosting"])
Expand Down
23 changes: 13 additions & 10 deletions examples/ibis/table_dataflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,17 +24,20 @@ def feature_table(raw_table: ir.Table) -> ir.Table:
is_summer_brazil=ibis._.month_of_absence.isin([1, 2, 12]),
)


@check_output(
schema=ibis.schema({
'has_children': "int",
'has_pet': "bool",
'is_summer_brazil': "bool",
'service_time': "int",
'seasons': "int",
'disciplinary_failure': "int",
'absenteeism_time_in_hours': "int",
}),
importance="fail"
schema=ibis.schema(
{
"has_children": "int",
"has_pet": "bool",
"is_summer_brazil": "bool",
"service_time": "int",
"seasons": "int",
"disciplinary_failure": "int",
"absenteeism_time_in_hours": "int",
}
),
importance="fail",
)
def feature_set(
feature_table: ir.Table,
Expand Down

0 comments on commit 85d7554

Please sign in to comment.