diff --git a/recipes_source/recipes/benchmark.py b/recipes_source/recipes/benchmark.py index 96fdf109bd..de50e0dc55 100644 --- a/recipes_source/recipes/benchmark.py +++ b/recipes_source/recipes/benchmark.py @@ -553,7 +553,7 @@ def batched_dot_bmm(a, b): # cases for the benchmark. # -from torch.utils.benchmark import Fuzzer, FuzzedParameter, FuzzedTensor, ParameterAlias +from torch.utils.benchmark import Fuzzer, FuzzedParameter, FuzzedTensor # Generates random tensors with 128 to 10000000 elements and sizes k0 and k1 chosen from a # ``loguniform`` distribution in [1, 10000], 40% of which will be discontiguous on average.