Skip to content

Commit 810e211

Browse files
Remove unused DynamicRetrievalServer and Bert dataset loader classes (#14209) (#14432)
* remove vulnerability code * Apply isort and black reformatting * fix import * fix style --------- Signed-off-by: dimapihtar <dpihtar@gmail.com> Signed-off-by: dimapihtar <dimapihtar@users.noreply.github.com> Signed-off-by: Charlie Truong <chtruong@nvidia.com> Co-authored-by: Dmytro Pykhtar <37850217+dimapihtar@users.noreply.github.com> Co-authored-by: dimapihtar <dimapihtar@users.noreply.github.com>
1 parent f98ef1d commit 810e211

File tree

6 files changed

+43
-664
lines changed

6 files changed

+43
-664
lines changed

nemo/collections/nlp/data/__init__.py

Lines changed: 15 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -12,34 +12,33 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from nemo.collections.nlp.data.data_utils import *
16-
from nemo.collections.nlp.data.entity_linking.entity_linking_dataset import EntityLinkingDataset
17-
from nemo.collections.nlp.data.information_retrieval.information_retrieval_dataset import (
15+
from nemo.collections.nlp.data.data_utils import * # noqa: F401
16+
from nemo.collections.nlp.data.entity_linking.entity_linking_dataset import EntityLinkingDataset # noqa: F401
17+
from nemo.collections.nlp.data.information_retrieval.information_retrieval_dataset import ( # noqa: F401
1818
BertInformationRetrievalDataset,
1919
)
20-
from nemo.collections.nlp.data.language_modeling.l2r_lm_dataset import (
20+
from nemo.collections.nlp.data.language_modeling.l2r_lm_dataset import ( # noqa: F401
2121
L2RLanguageModelingDataset,
2222
TarredL2RLanguageModelingDataset,
2323
)
24-
from nemo.collections.nlp.data.language_modeling.lm_bert_dataset import (
25-
BertPretrainingDataset,
26-
BertPretrainingPreprocessedDataloader,
24+
from nemo.collections.nlp.data.language_modeling.sentence_dataset import ( # noqa: F401
25+
SentenceDataset,
26+
TarredSentenceDataset,
2727
)
28-
from nemo.collections.nlp.data.language_modeling.sentence_dataset import SentenceDataset, TarredSentenceDataset
29-
from nemo.collections.nlp.data.machine_translation.machine_translation_dataset import (
28+
from nemo.collections.nlp.data.machine_translation.machine_translation_dataset import ( # noqa: F401
3029
TarredTranslationDataset,
3130
TranslationDataset,
3231
)
33-
from nemo.collections.nlp.data.question_answering_squad.qa_dataset import SquadDataset
34-
from nemo.collections.nlp.data.text2sparql.text2sparql_dataset import Text2SparqlDataset
35-
from nemo.collections.nlp.data.text_normalization.decoder_dataset import TextNormalizationDecoderDataset
36-
from nemo.collections.nlp.data.text_normalization.tagger_dataset import TextNormalizationTaggerDataset
37-
from nemo.collections.nlp.data.text_normalization.test_dataset import TextNormalizationTestDataset
38-
from nemo.collections.nlp.data.token_classification.token_classification_dataset import (
32+
from nemo.collections.nlp.data.question_answering_squad.qa_dataset import SquadDataset # noqa: F401
33+
from nemo.collections.nlp.data.text2sparql.text2sparql_dataset import Text2SparqlDataset # noqa: F401
34+
from nemo.collections.nlp.data.text_normalization.decoder_dataset import TextNormalizationDecoderDataset # noqa: F401
35+
from nemo.collections.nlp.data.text_normalization.tagger_dataset import TextNormalizationTaggerDataset # noqa: F401
36+
from nemo.collections.nlp.data.text_normalization.test_dataset import TextNormalizationTestDataset # noqa: F401
37+
from nemo.collections.nlp.data.token_classification.token_classification_dataset import ( # noqa: F401
3938
BertTokenClassificationDataset,
4039
BertTokenClassificationInferDataset,
4140
)
42-
from nemo.collections.nlp.data.zero_shot_intent_recognition.zero_shot_intent_dataset import (
41+
from nemo.collections.nlp.data.zero_shot_intent_recognition.zero_shot_intent_dataset import ( # noqa: F401
4342
ZeroShotIntentDataset,
4443
ZeroShotIntentInferenceDataset,
4544
)

nemo/collections/nlp/data/language_modeling/__init__.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,8 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from nemo.collections.nlp.data.language_modeling.l2r_lm_dataset import L2RLanguageModelingDataset
16-
from nemo.collections.nlp.data.language_modeling.lm_bert_dataset import (
17-
BertPretrainingDataset,
18-
BertPretrainingPreprocessedDataloader,
15+
from nemo.collections.nlp.data.language_modeling.l2r_lm_dataset import L2RLanguageModelingDataset # noqa: F401
16+
from nemo.collections.nlp.data.language_modeling.sentence_dataset import ( # noqa: F401
17+
SentenceDataset,
18+
TarredSentenceDataset,
1919
)
20-
from nemo.collections.nlp.data.language_modeling.sentence_dataset import SentenceDataset, TarredSentenceDataset

0 commit comments

Comments
 (0)