Skip to content

Clean up nemo-retriever dependencies#2183

Draft
colesmcintosh wants to merge 1 commit into
NVIDIA:mainfrom
colesmcintosh:cleanup-nemo-retriever-dependencies
Draft

Clean up nemo-retriever dependencies#2183
colesmcintosh wants to merge 1 commit into
NVIDIA:mainfrom
colesmcintosh:cleanup-nemo-retriever-dependencies

Conversation

@colesmcintosh
Copy link
Copy Markdown

Summary

  • add runtime dependencies used by shipped nemo_retriever code to the core project dependencies
  • move duplicated extra-only entries out of optional extras now covered by core dependencies
  • add a test extra for MoviePy while constraining it to the Pillow-compatible 1.x line
  • regenerate nemo_retriever/uv.lock

Closes #1392.

Notes

  • The issue refers to api/pyproject.toml; on current main, the relevant manifest is nemo_retriever/pyproject.toml.
  • I used opencv-python-headless rather than opencv-python because this project already suppresses opencv-python in uv overrides to avoid cv2 directory conflicts.
  • Some packages listed in the older issue body, such as redis, python-docx, python-pptx, minio, and pymilvus, are not imported by the current tree and were not added as unused runtime dependencies.

Validation

  • python -c "import pathlib,tomllib; tomllib.loads(pathlib.Path('nemo_retriever/pyproject.toml').read_text()); print('pyproject ok')"
  • custom dependency manifest assertion for core/test dependency placement
  • uvx --from uv uv lock --check --quiet from nemo_retriever/
  • git diff --check

Not Run

  • python -m pytest nemo_retriever/tests/test_nemotron_ocr_v2_nightly.py::test_local_extra_accepts_stable_ocr_2_and_newer_dev_releases could not start in the ambient Python because project dependencies such as pandas are not installed there.

Signed-off-by: Cole McIntosh <colemcintosh6@gmail.com>
@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented May 31, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[BUG]: Clean up dependencies in api/pyproject.toml

1 participant