Forgot benchmarks

pytorch · Jun 20, 2024 · 511ac4f · 511ac4f
1 parent b6d0b80
commit 511ac4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/samplers/benchmark_samplers.py b/benchmarks/samplers/benchmark_samplers.py
@@ -11,7 +11,7 @@
 import torch
 
 import torch.utils.benchmark as benchmark
-from torchcodec.samplers import (
+from torchcodec.samplers._video_clip_sampler import (
     IndexBasedSamplerArgs,
     TimeBasedSamplerArgs,
     VideoArgs,