Skip to content

Commit 373782f

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

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

kernel_tuner/strategies/bayes_opt_BOTorch_transfer_RGPE.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def __init__(self, searchspace: Searchspace, runner, tuning_options):
5252
self.outcomes_transfer_learning: list[Tensor] = []
5353
self.models_transfer_learning: list = []
5454
for tl_cache in tuning_options.transfer_learning_caches:
55-
print(f"Importing transfer learning for {tl_cache["kernel_name"]}-{tl_cache['device_name']}")
55+
print(f"Importing transfer learning for {tl_cache['kernel_name']}-{tl_cache['device_name']}")
5656
# construct the searchspace for this task
5757
tensor_kwargs = searchspace.tensor_kwargs
5858
tl_searchspace = Searchspace(None, None, None, from_cache=tl_cache)

kernel_tuner/strategies/bayes_opt_BOTorch_transfer_weighted.py

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

0 commit comments

Comments
 (0)