Skip to content

Commit 2034928

Browse files
fineguyThe TensorFlow Datasets Authors
authored andcommitted
Pin apache-beam to <2.65.0.
PiperOrigin-RevId: 760933928
1 parent d56f79d commit 2034928

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

setup.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,8 @@
8181
'pytest-shard',
8282
'pytest-xdist',
8383
# Lazy-deps required by core
84-
'apache-beam',
84+
# TODO(b/418761065): Update to 2.65.0 once the bug is fixed.
85+
'apache-beam<2.65.0',
8586
'conllu',
8687
'mlcroissant>=1.0.9',
8788
'pandas',
@@ -141,17 +142,17 @@
141142
DATASET_EXTRAS = {
142143
# In alphabetical order
143144
'aflw2k3d': ['scipy'],
144-
'beir': ['apache-beam'],
145+
'beir': ['apache-beam<2.65.0'],
145146
'ble_wind_field': ['gcsfs', 'zarr<3.0.0'],
146147
'c4': [
147-
'apache-beam',
148+
'apache-beam<2.65.0',
148149
'gcld3',
149150
'langdetect',
150151
# nltk==3.8.2 is broken: https://github.com/nltk/nltk/issues/3293
151152
'nltk==3.8.1',
152153
'tldextract',
153154
],
154-
'c4_wsrs': ['apache-beam'],
155+
'c4_wsrs': ['apache-beam<2.65.0'],
155156
'cats_vs_dogs': ['matplotlib'],
156157
'colorectal_histology': ['Pillow'],
157158
'common_voice': ['pydub'], # and ffmpeg installed
@@ -188,8 +189,8 @@
188189
'the300w_lp': ['scipy'],
189190
'wake_vision': ['pandas'],
190191
'wider_face': ['Pillow'],
191-
'wiki_dialog': ['apache-beam'],
192-
'wikipedia': ['apache-beam', 'mwparserfromhell', 'mwxml'],
192+
'wiki_dialog': ['apache-beam<2.65.0'],
193+
'wikipedia': ['apache-beam<2.65.0', 'mwparserfromhell', 'mwxml'],
193194
'wsc273': ['bs4', 'lxml'],
194195
'youtube_vis': ['pycocotools'],
195196
}

0 commit comments

Comments
 (0)