Skip to content

Commit

Permalink
Use test_installs to setup the dependencies
Browse files Browse the repository at this point in the history
  • Loading branch information
mfuntowicz committed Apr 3, 2024
1 parent 1706b36 commit bcf6080
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -69,19 +69,17 @@ pypi_upload: ${PACKAGE_DIST} ${PACKAGE_WHEEL}
python -m twine upload ${PACKAGE_DIST} ${PACKAGE_WHEEL}

# Tests

test_installs:
python -m pip install .[tests]
python -m pip install git+https://github.com/huggingface/transformers.git
python -m pip install .[tpu,tests]
# python -m pip install git+https://github.com/huggingface/transformers.git

# Stand-alone TGI server for unit tests outside of TGI container
tgi_server:
python -m pip install -r text-generation-inference/server/build-requirements.txt
make -C text-generation-inference/server clean
VERSION=${VERSION} TGI_VERSION=${TGI_VERSION} make -C text-generation-inference/server gen-server

tgi_test: tgi_server
python -m pip install .[tpu,tests]
tgi_test: test_installs tgi_server
find text-generation-inference -name "text_generation_server-$(VERSION)-py3-none-any.whl" \
-exec python -m pip install --force-reinstall {} \;
python -m pytest -sv text-generation-inference/tests
Expand Down

0 comments on commit bcf6080

Please sign in to comment.