|
81 | 81 | 'pytest-shard',
|
82 | 82 | 'pytest-xdist',
|
83 | 83 | # 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', |
85 | 86 | 'conllu',
|
86 | 87 | 'mlcroissant>=1.0.9',
|
87 | 88 | 'pandas',
|
|
141 | 142 | DATASET_EXTRAS = {
|
142 | 143 | # In alphabetical order
|
143 | 144 | 'aflw2k3d': ['scipy'],
|
144 |
| - 'beir': ['apache-beam'], |
| 145 | + 'beir': ['apache-beam<2.65.0'], |
145 | 146 | 'ble_wind_field': ['gcsfs', 'zarr<3.0.0'],
|
146 | 147 | 'c4': [
|
147 |
| - 'apache-beam', |
| 148 | + 'apache-beam<2.65.0', |
148 | 149 | 'gcld3',
|
149 | 150 | 'langdetect',
|
150 | 151 | # nltk==3.8.2 is broken: https://github.com/nltk/nltk/issues/3293
|
151 | 152 | 'nltk==3.8.1',
|
152 | 153 | 'tldextract',
|
153 | 154 | ],
|
154 |
| - 'c4_wsrs': ['apache-beam'], |
| 155 | + 'c4_wsrs': ['apache-beam<2.65.0'], |
155 | 156 | 'cats_vs_dogs': ['matplotlib'],
|
156 | 157 | 'colorectal_histology': ['Pillow'],
|
157 | 158 | 'common_voice': ['pydub'], # and ffmpeg installed
|
|
188 | 189 | 'the300w_lp': ['scipy'],
|
189 | 190 | 'wake_vision': ['pandas'],
|
190 | 191 | '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'], |
193 | 194 | 'wsc273': ['bs4', 'lxml'],
|
194 | 195 | 'youtube_vis': ['pycocotools'],
|
195 | 196 | }
|
|
0 commit comments