Skip to content
15 changes: 1 addition & 14 deletions python/cudf_polars/cudf_polars/experimental/benchmarks/pdsds.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,6 @@
COUNT_DTYPE,
build_parser,
parse_args,
run_duckdb,
run_polars,
)
except ImportError as e:
Expand Down Expand Up @@ -328,17 +327,5 @@ class PDSDSDuckDBQueries(PDSDSQueries):

if __name__ == "__main__":
parser = build_parser(num_queries=99)
parser.add_argument(
"--engine",
choices=["polars", "duckdb"],
default="polars",
help="Which engine to use for executing the benchmarks or to validate results.",
)
args = parse_args(parser=parser)

if args.engine == "polars":
run_polars(PDSDSPolarsQueries, args)
elif args.engine == "duckdb":
run_duckdb(PDSDSDuckDBQueries, args)
else:
raise ValueError(f"Invalid engine: {args.engine}")
run_polars(PDSDSPolarsQueries, args)
15 changes: 1 addition & 14 deletions python/cudf_polars/cudf_polars/experimental/benchmarks/pdsh.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,6 @@
build_parser,
get_data,
parse_args,
run_duckdb,
run_polars,
)
except ImportError as e:
Expand Down Expand Up @@ -1795,17 +1794,5 @@ def q22(run_config: RunConfig) -> str:

if __name__ == "__main__":
parser = build_parser(num_queries=22)
parser.add_argument(
"--engine",
choices=["polars", "duckdb"],
default="polars",
help="Which engine to use for executing the benchmarks or to validate results.",
)
args = parse_args(parser=parser)

if args.engine == "polars":
run_polars(PDSHQueries, args)
elif args.engine == "duckdb":
run_duckdb(PDSHDuckDBQueries, args)
else:
raise ValueError(f"Invalid engine: {args.engine}")
run_polars(PDSHQueries, args)
Loading
Loading