Skip to content

Commit 31ba0dc

Browse files
radekosmulskikarlhigleybschifferer
authored
MM transformer training and serving example (#1045)
Add inference to the next item prediction example (session-based) * extend example * align unit test * add merlin-systems to envs --------- Co-authored-by: Karl Higley <[email protected]> Co-authored-by: Benedikt Schifferer <[email protected]>
1 parent 2a20547 commit 31ba0dc

File tree

4 files changed

+644
-150
lines changed

4 files changed

+644
-150
lines changed

.github/workflows/tensorflow.yml

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,14 @@ jobs:
4747
fi
4848
pip install "pandas>=1.2.0,<1.4.0dev0"
4949
pip install "NVTabular@git+https://github.com/NVIDIA-Merlin/NVTabular.git@$branch"
50-
pip install "merlin-dataloader@git+https://github.com/NVIDIA-Merlin/dataloader.git@$branch"
50+
pip install "merlin-dataloader@git+https://github.com/NVIDIA-Merlin/dataloader.git@$branch"
5151
pip install "merlin-core@git+https://github.com/NVIDIA-Merlin/core.git@$branch"
52+
pip install "merlin-systems@git+https://github.com/NVIDIA-Merlin/systems.git@$branch"
5253
- name: Install dependencies
5354
run: |
5455
python -m pip install "tensorflow${{ matrix.tensorflow-version }}"
55-
python -m pip install .[tensorflow-dev]
56+
python -m pip install .[tensorflow-dev]
57+
pip install tritonclient[all]
5658
- name: Build
5759
run: |
5860
python setup.py develop
@@ -102,12 +104,14 @@ jobs:
102104
fi
103105
pip install "pandas>=1.2.0,<1.4.0dev0"
104106
pip install "NVTabular@git+https://github.com/NVIDIA-Merlin/NVTabular.git@$branch"
105-
pip install "merlin-dataloader@git+https://github.com/NVIDIA-Merlin/dataloader.git@$branch"
107+
pip install "merlin-dataloader@git+https://github.com/NVIDIA-Merlin/dataloader.git@$branch"
106108
pip install "merlin-core@git+https://github.com/NVIDIA-Merlin/core.git@$branch"
109+
pip install "merlin-systems@git+https://github.com/NVIDIA-Merlin/systems.git@$branch"
107110
- name: Install dependencies
108111
run: |
109112
python -m pip install "tensorflow${{ matrix.tensorflow-version }}"
110-
python -m pip install .[tensorflow-dev]
113+
python -m pip install .[tensorflow-dev]
114+
pip install tritonclient[all]
111115
- name: Build
112116
run: |
113117
python setup.py develop

0 commit comments

Comments
 (0)