From 18a5dce7b1f7f50a6e5b52570f615f22f1c02633 Mon Sep 17 00:00:00 2001
From: Liam Connors <connorsl@tcd.ie>
Date: Tue, 1 Oct 2024 03:41:40 -0400
Subject: [PATCH] update check for cuDF series (#1105)

---
 tests/utils.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/utils.py b/tests/utils.py
index b2030744c..15ce25140 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -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
@@ -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]):