From a6251e0d94396674ad7c376058bfb784a5a97a15 Mon Sep 17 00:00:00 2001
From: Han Wang <Han.Wang@amd.com>
Date: Fri, 22 Nov 2024 07:23:28 +0000
Subject: [PATCH] test: passing `device` in str type

---
 tests/exporters/onnx/test_exporters_onnx_cli.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/exporters/onnx/test_exporters_onnx_cli.py b/tests/exporters/onnx/test_exporters_onnx_cli.py
index 9701f3b390..df984a0a08 100644
--- a/tests/exporters/onnx/test_exporters_onnx_cli.py
+++ b/tests/exporters/onnx/test_exporters_onnx_cli.py
@@ -846,7 +846,7 @@ def replace_activation(model: nn.Module):
                         monolith,
                         no_post_process,
                         variant=variant,
-                        device=model.device,
+                        device=device,
                         disable_dynamic_axes_fix=not do_validation,
                         do_validation=do_validation,
                         custom_export_fn=custom_export_mock,
@@ -859,7 +859,7 @@ def replace_activation(model: nn.Module):
                 monolith,
                 no_post_process,
                 variant=variant,
-                device=model.device,
+                device=device,
                 disable_dynamic_axes_fix=True,
                 do_validation=False,
                 custom_export_fn=custom_export_mock,
@@ -873,7 +873,7 @@ def replace_activation(model: nn.Module):
                 monolith,
                 no_post_process,
                 variant=variant,
-                device=model.device,
+                device=device,
                 disable_dynamic_axes_fix=not do_validation,
                 do_validation=do_validation,
                 custom_export_fn=custom_export_mock,