Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -28,3 +28,6 @@ pytest-pythonpath
# Coverage statistics
pytest-cov
codecov

# To parse untrusted XML data
defusedxml
5 changes: 4 additions & 1 deletion torchtext/data/datasets_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,10 @@
unicode_csv_reader,
)
import codecs
import xml.etree.ElementTree as ET
try:
import defusedxml.ElementTree as ET
except ImportError:
import xml.etree.ElementTree as ET
"""
These functions and classes are meant solely for use in torchtext.datasets and not
for public consumption yet.
Expand Down
5 changes: 4 additions & 1 deletion torchtext/legacy/datasets/translation.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@
import os
import xml.etree.ElementTree as ET
try:
import defusedxml.ElementTree as ET
except ImportError:
import xml.etree.ElementTree as ET
import glob
import io
import codecs
Expand Down