fixing function inspecting for torch

ivy-llc · Sep 13, 2023 · ccbf232 · ccbf232
1 parent d2e6cea
commit ccbf232
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ivy/functional/backends/torch/device.py b/ivy/functional/backends/torch/device.py
@@ -112,7 +112,7 @@ def handle_soft_device_variable(*args, fn, **kwargs):
     )
     # checking if this function accepts `device` argument
     # must be handled in the backend
-    if "device" in inspect.getfullargspec(fn).args:
+    if "device" in inspect.signature(fn).parameters:
         kwargs["device"] = device_shifting_dev
     return fn(*args, **kwargs)
 

diff --git a/ivy/functional/ivy/experimental/manipulation.py b/ivy/functional/ivy/experimental/manipulation.py
@@ -2638,7 +2638,7 @@ def choose(
 @inputs_to_ivy_arrays
 @handle_nestable
 @handle_exceptions
-@handle_device_shifting
+@handle_device
 def column_stack(
     arrays: Sequence[Union[ivy.Array, ivy.NativeArray]],
     /,