fix

Author: NeoLegends

tests/test_torch_dataset.py

@@ -203,16 +203,14 @@ def test_MultiProcDataset_HDFDataset():
 
 
 def test_dataset_num_workers_sharding():
-    from returnn.datasets.audio import OggZipDataset
-
     config = Config({"backend": "torch", "torch_dataloader_opts": {"num_workers": 2}})
     with global_config_ctx(config):
         datasets = [
             init_dataset({"class": "Task12AXDataset", "num_seqs": 100, "num_shards": 2, "shard_index": i})
             for i in range(2)
         ]
         for dataset in datasets:
-            assert isinstance(dataset, OggZipDataset)
+            assert isinstance(dataset, Task12AXDataset)
             dataset.init_seq_order(epoch=1)
             assert dataset.shard_index < dataset.num_shards == 4
             assert dataset.num_seqs == 25