Skip to content

Commit

Permalink
Fix Dataset imports and module loading
Browse files Browse the repository at this point in the history
 - remove unused imports in generating.py
 - add stream_hdf to loaded module list
  • Loading branch information
JackTemaki committed Mar 28, 2024
1 parent 305c68f commit 2edf897
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 9 deletions.
7 changes: 5 additions & 2 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,11 @@

## Current Master (0.4+git)

- Remove everything related to RASR/Sprint interfacing
- Remove task-system that is no longer needed
- Remove "Sprint" related code https://github.com/JackTemaki/MiniReturnn/pull/17
- Remove everything related to RASR/Sprint interfacing
- Remove task-system that is no longer needed
- Remove Sprint Datasets
- Cleanup dataset imports and fix missing stream_hdf module loading


## Version 0.4
Expand Down
2 changes: 1 addition & 1 deletion returnn/datasets/basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -870,7 +870,7 @@ def get_dataset_class(name: Union[str, Type[Dataset]]) -> Optional[Type[Dataset]

# Only those modules which make sense to be loaded by the user,
# because this function is only used for such cases.
mod_names = ["hdf", "generating", "numpy_dump", "meta", "lm", "map"]
mod_names = ["audio", "generating", "hdf", "lm", "map", "meta", "numpy_dump", "stream_hdf"]
for mod_name in mod_names:
mod = import_module("returnn.datasets.%s" % mod_name)
for name_, clazz in vars(mod).items():
Expand Down
7 changes: 1 addition & 6 deletions returnn/datasets/generating.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,10 @@

from typing import Optional, Sequence
import numpy
import sys
import typing

from returnn.util.basic import class_idx_seq_to_1_of_k, CollectionReadCheckCovered
from returnn.log import log
from returnn.util.basic import class_idx_seq_to_1_of_k

from .util.feature_extraction import ExtractAudioFeatures
from .util.vocabulary import *
from .audio import OggZipDataset # noqa # for API compatibility
from .basic import Dataset, DatasetSeq, convert_data_dims
from .cached2 import CachedDataset2

Expand Down

0 comments on commit 2edf897

Please sign in to comment.