Skip to content

Commit 388f325

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

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kernel_tuner/strategies/bayes_opt_BOTorch_transfer_direct.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def __init__(self, searchspace: Searchspace, runner, tuning_options):
3535
self.inputs_transfer_learning: list[Tensor] = []
3636
self.outcomes_transfer_learning: list[Tensor] = []
3737
for tl_cache in tuning_options.transfer_learning_caches:
38-
print(f"Importing transfer learning for {tl_cache["kernel_name"]}-{tl_cache['device_name']}")
38+
print(f"Importing transfer learning for {tl_cache['kernel_name']}-{tl_cache['device_name']}")
3939
# construct the searchspace for this task
4040
tensor_kwargs = searchspace.tensor_kwargs
4141
tl_searchspace = Searchspace(None, None, None, from_cache=tl_cache)

0 commit comments

Comments
 (0)