Skip to content

Commit

Permalink
update check for cuDF series (#1105)
Browse files Browse the repository at this point in the history
  • Loading branch information
LiamConnors authored Oct 1, 2024
1 parent 9fb12be commit 18a5dce
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions tests/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
import pandas as pd

from narwhals.typing import IntoFrame
from narwhals.utils import Implementation

if sys.version_info >= (3, 10):
from typing import TypeAlias # pragma: no cover
Expand All @@ -35,8 +36,9 @@ def compare_dicts(result: Any, expected: dict[str, Any]) -> None:
assert key in expected
for key in expected:
result_key = result[key]
if hasattr(result_key, "_compliant_series") and "CUDF" in str(
result_key._compliant_series._implementation
if (
hasattr(result_key, "_compliant_series")
and result_key._compliant_series._implementation is Implementation.CUDF
): # pragma: no cover
result_key = result_key.to_pandas()
for lhs, rhs in zip_strict(result_key, expected[key]):
Expand Down

0 comments on commit 18a5dce

Please sign in to comment.