|
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