Skip to content

Commit 3c48b49

Browse files
committed
Fixed torch import error due to Tensor type hint
1 parent 435b56b commit 3c48b49

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kernel_tuner/searchspace.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -707,7 +707,7 @@ def param_config_to_tensor(self, param_config: tuple):
707707
raise KeyError(f"No variant of {param} could be found in {mapping}") from e
708708
return torch.tensor(array, **self.tensor_kwargs)
709709

710-
def tensor_to_param_config(self, tensor: Tensor):
710+
def tensor_to_param_config(self, tensor):
711711
"""Convert from a Tensor to a parameter configuration."""
712712
assert tensor.dim() == 1, f"Parameter configuration tensor must be 1-dimensional, is {tensor.dim()} ({tensor})"
713713
if len(self._map_tensor_to_param) == 0:

0 commit comments

Comments
 (0)