From 85d755456700625daac0002a1368a565b03935db Mon Sep 17 00:00:00 2001 From: zilto Date: Sun, 3 Mar 2024 13:42:14 -0500 Subject: [PATCH] pre-commit hooks --- examples/ibis/run.py | 2 +- examples/ibis/table_dataflow.py | 23 +++++++++++++---------- 2 files changed, 14 insertions(+), 11 deletions(-) diff --git a/examples/ibis/run.py b/examples/ibis/run.py index f5fed9a03..de71bcf9b 100644 --- a/examples/ibis/run.py +++ b/examples/ibis/run.py @@ -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"]) diff --git a/examples/ibis/table_dataflow.py b/examples/ibis/table_dataflow.py index b3678601f..c668112be 100644 --- a/examples/ibis/table_dataflow.py +++ b/examples/ibis/table_dataflow.py @@ -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,