# Please keep this in alphabetical order
Boruta
Cartopy
ImageHash
Janome
PyArabic
PyUpSet
Pympler
Rtree
shapely
SimpleITK
TPOT
Wand
bayesian-optimization
boto3
catboost
category-encoders
cesium
comm
cytoolz
# Older versions of datasets fail with "Loading a dataset cached in a LocalFileSystem is not supported"
# https://stackoverflow.com/questions/77433096/notimplementederror-loading-a-dataset-cached-in-a-localfilesystem-is-not-suppor
datasets>=2.14.6
deap
dipy
docker
easyocr
emoji
fasttext
featuretools
fiona
flax>=0.12
# TODO: b/545164527 - unpin once dipy ships its fury<2.0 cap.
# fury 2.0 swapped VTK for pygfx/wgpu, breaking `from dipy.viz import window`.
# dipy requires fury uncapped; fix is merged upstream but unreleased:
# https://github.com/dipy/dipy/issues/3978
fury<2
fuzzywuzzy
geojson
gensim
# b/443054743,b/455550872
google-adk[a2a,eval]>=1.21.0
google-cloud-aiplatform
google-cloud-videointelligence
google-cloud-vision
google-genai
gpxpy
h2o
haversine
hep-ml
igraph
ipympl
ipywidgets==8.1.5
isoweek
jedi
# jitler 0.11.1 breaks simulation image
jiter==0.10.0
# b/276358430: jupyter-lsp was pinned to 1.5.1 to stop Jupyter lsp freezing up
# the jupyter server. That pin is unusable on Python 3.13: it holds jupyterlab
# back to the 3.x line, which pulls jupyter-ydoc 0.2.5 -> y-py 0.6.2, and y-py
# publishes no cp313 wheel and its sdist no longer builds. jupyterlab 4.x
# replaced y-py with pycrdt, so leave jupyter-lsp unpinned.
# b/333854354: pin jupyter-server to version 2.12.5; later versions break LSP (b/333854354)
jupyter_server==2.12.5
jupyter_server_proxy
jupyterlab
jupyterlab-lsp
kaggle>=1.8.3
kaggle-environments
kagglehub[pandas-datasets,hf-datasets,signing]>=0.4.2
keras-cv
keras-nlp
keras-tuner
kornia
langid
libpysal
lime
line_profiler
mamba
matplotlib
mlcrate
mne
mpld3
# b/274619697: learntools requires a specific nbconvert right now
nbconvert==6.4.5
nbdev
nilearn
olefile
onnx
# The base image ships an inconsistent OpenCV set: opencv-python and
# opencv-python-headless at 5.0.0.93 but opencv-contrib-python at 4.14.0.94.
# They all install into the same cv2/ package and clobber each other, and
# OpenCV 5.0 dropped long-standing APIs (e.g. cv2.CascadeClassifier). Hold the
# whole set on the last 4.x release so cv2 stays consistent and notebooks that
# use the 4.x API keep working.
opencv-contrib-python<5
opencv-python<5
opencv-python-headless<5
openslide-bin
openslide-python
optuna
pandas-profiling
pandasql
papermill
path
path.py
pdf2image
# The base image ships pillow 11.3; pin up to a newer release.
# Remove once the base image ships >=12.2 itself.
pillow>=12.2
plotly-express
pudb
pyLDAvis
pycryptodome
pydicom
pyemd
pyexcel-ods
pymongo
pypdf
pytesseract
python-lsp-server
pytorch-ignite
pytorch-lightning
qgrid
qtconsole
ray
rgf-python
s3fs
scikit-learn
scikit-multilearn
scikit-optimize
scikit-plot
scikit-surprise
git+https://github.com/facebookresearch/segment-anything.git
squarify
# tensorflow-io was dropped when the base image moved to Python 3.13. Its last
# release (0.37.1, 2024-07-01) declares requires_python <3.13 and ships no cp313
# wheels, and the project is unmaintained.
# Must be compatible with torch version: https://github.com/meta-pytorch/torchcodec?tab=readme-ov-file#installing-torchcodec
torchcodec==0.11.0
torchinfo
torchmetrics
torchtune
transformers>=5.0.0
# The base image ships urllib3 2.5; pin up to a newer release.
# Remove once the base image ships >=2.7 itself.
urllib3>=2.7
universal-pathlib
vtk
wavio
xvfbwrapper
ydata-profiling