Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Include requirements files in the morpheus packages #1957

Merged
merged 6 commits into from
Oct 19, 2024
Merged
Show file tree
Hide file tree
Changes from 5 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions ci/conda/recipes/morpheus-libs/morpheus_core_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,11 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# morpheus tests are dependent on some pypi packages
rapids-dependency-file-generator \
--output requirements \
--file-key morpheus_core --matrix "" | tee "core_requirements.txt"

pip install -r core_requirements.txt
python3 <<EOF
import importlib.resources
import subprocess
requirements_file = importlib.resources.path("morpheus", "requirements_morpheus_core.txt")
subprocess.call(f"pip install -r {requirements_file}".split())
EOF

pytest tests/morpheus
12 changes: 6 additions & 6 deletions ci/conda/recipes/morpheus-libs/morpheus_dfp_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,11 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# morpheus tests are dependent on some pypi packages
rapids-dependency-file-generator \
--output requirements \
--file-key morpheus_dfp --matrix "" | tee "dfp_requirements.txt"

pip install -r dfp_requirements.txt
python3 <<EOF
import importlib.resources
import subprocess
requirements_file = importlib.resources.path("morpheus_dfp", "requirements_morpheus_dfp.txt")
subprocess.call(f"pip install -r {requirements_file}".split())
EOF

pytest tests/morpheus_dfp
13 changes: 7 additions & 6 deletions ci/conda/recipes/morpheus-libs/morpheus_llm_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,12 @@
# See the License for the specific language governing permissions and
# limitations under the License.

# llm library and tests are dependent on a number of pypi packages - fixme
rapids-dependency-file-generator \
--output requirements \
--file-key morpheus_llm --matrix "" | tee "llm_requirements.txt"

pip install -r llm_requirements.txt
# Install requirements if they are included in the package
python3 <<EOF
import importlib.resources
import subprocess
requirements_file = importlib.resources.path("morpheus_llm", "requirements_morpheus_llm.txt")
subprocess.call(f"pip install -r {requirements_file}".split())
EOF

pytest tests/morpheus_llm
37 changes: 20 additions & 17 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -176,24 +176,27 @@ files:
- checks
- python

# pip dependencies that are used during the conda test stage for morpheus-llm output
morpheus_llm:
output: none
includes:
- test_morpheus_llm_pip
- test_morpheus_core_pip

# pip dependencies that are used during the conda test stage for morpheus-core output
# morpheus-core pip dependencies are published as a morpheus package data file
morpheus_core:
output: none
output: requirements
requirements_dir: python/morpheus/morpheus
includes:
- test_morpheus_core_pip
- morpheus_core_pip

# pip dependencies that are used during the conda test stage for morpheus-dfp output
# morpheus-dfp pip dependencies are published as a morpheus_dfp package data file
morpheus_dfp:
output: none
output: requirements
requirements_dir: python/morpheus_dfp/morpheus_dfp
includes:
- test_morpheus_core_pip
- morpheus_core_pip

# morpheus-llm pip dependencies are published as a morpheus_llm package data file
morpheus_llm:
output: requirements
requirements_dir: python/morpheus_llm/morpheus_llm
includes:
- morpheus_core_pip
- morpheus_llm_pip

channels:
- conda-forge
Expand Down Expand Up @@ -402,24 +405,24 @@ dependencies:
- pip:
- pytest-kafka==0.6.0

test_morpheus_core_pip:
morpheus_core_pip:
common:
- output_types: [requirements]
packages:
- --extra-index-url https://download.pytorch.org/whl/cu124
- torch==2.4.0+cu124

test_morpheus_llm_pip:
morpheus_llm_pip:
common:
- output_types: [requirements]
packages:
- faiss-cpu
AnuradhaKaruppiah marked this conversation as resolved.
Show resolved Hide resolved
- milvus==2.3.5 # update to match pymilvus when available
- pymilvus==2.3.6
- langchain==0.1.16
- langchain-nvidia-ai-endpoints==0.0.11
- faiss-cpu
- google-search-results==2.4
- nemollm==0.3.5
- nemollm==0.3.5

example-dfp-prod:
common:
Expand Down
3 changes: 3 additions & 0 deletions python/morpheus/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,9 @@ endif()
file(GLOB morpheus_data_files "${CMAKE_CURRENT_SOURCE_DIR}/morpheus/data/*")
morpheus_utils_add_python_sources(${morpheus_data_files})

file(GLOB morpheus_requirement_files "${CMAKE_CURRENT_SOURCE_DIR}/morpheus/requirements*")
morpheus_utils_add_python_sources(${morpheus_requirement_files})

morpheus_utils_build_python_package(morpheus ${extra_args})

list(POP_BACK CMAKE_MESSAGE_CONTEXT)
1 change: 1 addition & 0 deletions python/morpheus/MANIFEST.in
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
include morpheus/_version.py
include morpheus/requirements_morpheus_core.txt
recursive-include morpheus/data *
recursive-include morpheus *.cpython*.so py.typed *.pyi
4 changes: 4 additions & 0 deletions python/morpheus/morpheus/requirements_morpheus_core.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# This file is generated by `rapids-dependency-file-generator`.
# To make changes, edit ../../../dependencies.yaml and run `rapids-dependency-file-generator`.
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.4.0+cu124
3 changes: 3 additions & 0 deletions python/morpheus_dfp/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,9 @@ if(MORPHEUS_PYTHON_PERFORM_INSTALL)
list(APPEND extra_args "INSTALL_WHEEL")
endif()

file(GLOB morpheus_dfp_requirement_files "${CMAKE_CURRENT_SOURCE_DIR}/morpheus_dfp/requirements*")
morpheus_utils_add_python_sources(${morpheus_dfp_requirement_files})

morpheus_utils_build_python_package(morpheus_dfp ${extra_args})

list(POP_BACK CMAKE_MESSAGE_CONTEXT)
1 change: 1 addition & 0 deletions python/morpheus_dfp/MANIFEST.in
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
include morpheus_dfp/_version.py
include morpheus_dfp/requirements_morpheus_dfp.txt
recursive-include morpheus_dfp py.typed
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
# This file is generated by `rapids-dependency-file-generator`.
# To make changes, edit ../../../dependencies.yaml and run `rapids-dependency-file-generator`.
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.4.0+cu124
3 changes: 3 additions & 0 deletions python/morpheus_llm/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,9 @@ if(MORPHEUS_PYTHON_PERFORM_INSTALL)
list(APPEND extra_args "INSTALL_WHEEL")
endif()

file(GLOB morpheus_llm_requirement_files "${CMAKE_CURRENT_SOURCE_DIR}/morpheus_llm/requirements*")
morpheus_utils_add_python_sources(${morpheus_llm_requirement_files})

morpheus_utils_build_python_package(morpheus_llm ${extra_args})

list(POP_BACK CMAKE_MESSAGE_CONTEXT)
1 change: 1 addition & 0 deletions python/morpheus_llm/MANIFEST.in
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
include morpheus_llm/_version.py
include morpheus_llm/requirements_morpheus_llm.txt
recursive-include morpheus_llm *.cpython*.so py.typed *.pyi
11 changes: 11 additions & 0 deletions python/morpheus_llm/morpheus_llm/requirements_morpheus_llm.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
# This file is generated by `rapids-dependency-file-generator`.
# To make changes, edit ../../../dependencies.yaml and run `rapids-dependency-file-generator`.
--extra-index-url https://download.pytorch.org/whl/cu124
faiss-cpu
AnuradhaKaruppiah marked this conversation as resolved.
Show resolved Hide resolved
google-search-results==2.4
langchain-nvidia-ai-endpoints==0.0.11
langchain==0.1.16
milvus==2.3.5
nemollm==0.3.5
pymilvus==2.3.6
torch==2.4.0+cu124
Loading