Skip to content

Commit

Permalink
Fix misleading runtime erro messafe, remove unused inferring gpu mode…
Browse files Browse the repository at this point in the history
… based on the use_cudf marker
  • Loading branch information
dagardner-nv committed Oct 17, 2024
1 parent a0f5e5b commit 80e817a
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -209,7 +209,7 @@ def df_type(request: pytest.FixtureRequest):
use_cudf = request.node.get_closest_marker("use_cudf") is not None

if (use_pandas and use_cudf):
raise RuntimeError(f"Both markers (gpu_mode and cpu_mode) were added to function {request.node.nodeid}. "
raise RuntimeError(f"Both markers (use_pandas and use_cudf) were added to function {request.node.nodeid}. "
"Remove markers to support both.")

# This will default to "cudf" or follow use_pandas
Expand All @@ -235,7 +235,6 @@ def _get_execution_mode(request: pytest.FixtureRequest) -> "ExecutionMode":

# if both are undefined, infer based on the df_type
if (not gpu_mode and not cpu_mode):
gpu_mode = request.node.get_closest_marker("use_cudf") is not None
cpu_mode = request.node.get_closest_marker("use_pandas") is not None

# This will default to True or follow gpu_mode
Expand Down

0 comments on commit 80e817a

Please sign in to comment.