From dc763bbbe7352fed26d80af0d879f57208d3073e Mon Sep 17 00:00:00 2001 From: Chukwuma Nwaugha <20521315+nwaughachukwuma@users.noreply.github.com> Date: Thu, 11 Dec 2025 18:41:08 +0000 Subject: [PATCH] Update tests/conftest.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Signed-off-by: Chukwuma Nwaugha <20521315+nwaughachukwuma@users.noreply.github.com> --- tests/conftest.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/conftest.py b/tests/conftest.py index e4c2384e874f1..2ec5bd4f43a1b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -265,7 +265,7 @@ class HfRunner: if isinstance(x, dict): return {k: self.wrap_device(v, device) for k, v in x.items()} - if hasattr(x.device, "type") and x.device.type == device: + if hasattr(x, "device") and x.device.type == device: return x return x.to(device)