# Base deps for notebooks in train_scripts/.
#
# Install order for this repo:
#   python -m pip install -r deploy/requirements/nn_gpu_pinned.txt
#   python -m pip install -r train_scripts/requirements-train.txt
#   python -m pip install -e ./torchsig --no-deps
#
# Notes:
# - Do not install `torchsig` from PyPI: the project uses the local repo copy.
# - Main NN service pins come from deploy/requirements/nn_common.txt.
# - For notebooks use GUI OpenCV, so this file uses `opencv-python`, not headless.

ipykernel
jupyterlab
numpy==2.1.3
matplotlib==3.10.0
tqdm==4.67.1
PyYAML==6.0.2
mlconfig==0.3.2
scikit-learn==1.6.0
opencv-python==4.10.0.84
pandas
Pillow
scipy