asvdb tracking for next item prediction (#1042)

* add initial test under tests/benchmark --------- Co-authored-by: edknv <[email protected]>
NVIDIA-Merlin · Jun 12, 2023 · ed45657 · ed45657
1 parent ede0ef8
commit ed45657
Show file tree

Hide file tree

Showing 6 changed files with 333 additions and 205 deletions.
diff --git a/.github/workflows/tensorflow.yml b/.github/workflows/tensorflow.yml
@@ -74,6 +74,7 @@ jobs:
           python -m pip install "tensorflow${{ matrix.tensorflow-version }}"
           python -m pip install ".[tensorflow-dev]"
           pip install "tritonclient[all]"
+          pip install git+https://github.com/rapidsai/asvdb.git
       - name: Build
         run: |
           python setup.py develop
@@ -134,6 +135,7 @@ jobs:
           python -m pip install "tensorflow${{ matrix.tensorflow-version }}"
           python -m pip install ".[tensorflow-dev]"
           pip install "tritonclient[all]"
+          pip install git+https://github.com/rapidsai/asvdb.git
       - name: Build
         run: |
           python setup.py develop