diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless/asr_datamodule.py b/egs/wenetspeech/ASR/pruned_transducer_stateless/asr_datamodule.py
index 9941dc158..3c28dfa56 100644
--- a/egs/wenetspeech/ASR/pruned_transducer_stateless/asr_datamodule.py
+++ b/egs/wenetspeech/ASR/pruned_transducer_stateless/asr_datamodule.py
@@ -34,14 +34,12 @@ from lhotse.cut import Cut
 from lhotse.dataset import (
     CutConcatenate,
     CutMix,
-    BucketingSampler,
     DynamicBucketingSampler,
     K2SpeechRecognitionDataset,
     PrecomputedFeatures,
     SingleCutSampler,
     SpecAugment,
 )
-from lhotse.dataset.webdataset import export_to_webdataset
 from lhotse.dataset.input_strategies import OnTheFlyFeatures
 from torch.utils.data import DataLoader
 
@@ -365,6 +363,7 @@ class WenetSpeechAsrDataModule:
         )
 
         from lhotse.dataset.iterable_dataset import IterableDatasetWrapper
+
         test_iter_dataset = IterableDatasetWrapper(
             dataset=test,
             sampler=sampler,