Update tripy/nvtripy/backend/api/executable.py

yizhuoz004 · pranavm-nvidia · web-flow · commit 3e4b00ee667d · 2025-07-23T13:17:51.000-07:00
Co-authored-by: pranavm-nvidia &lt;49246958+pranavm-nvidia@users.noreply.github.com&gt;
Signed-off-by: yizhuoz004 &lt;yizhuoz@nvidia.com&gt;
diff --git a/tripy/nvtripy/backend/api/executable.py b/tripy/nvtripy/backend/api/executable.py
@@ -204,7 +204,7 @@ def add(a, b):
                 raise_error(
                     "Unexpected tensor device.",
                     [
-                        f"For tensor: `{arg_name}`, expected to be on device: {expected_device} but got: {tensor.device.kind}.\n",
+                        f"Expected tensor: `{arg_name}` to be on device: {expected_device} but it was on: {tensor.device.kind}.\n",
                     ],
                 )
 

Original file line number	Diff line number	Diff line change
`@@ -204,7 +204,7 @@ def add(a, b):`
`204`	`204`	`raise_error(`
`205`	`205`	`"Unexpected tensor device.",`
`206`	`206`	`[`
`207`		- f"For tensor: `{arg_name}`, expected to be on device: {expected_device} but got: {tensor.device.kind}.\n",
	`207`	+ f"Expected tensor: `{arg_name}` to be on device: {expected_device} but it was on: {tensor.device.kind}.\n",
`208`	`208`	`],`
`209`	`209`	`)`
`210`	`210`