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

Update test script #2995

Merged
merged 3 commits into from
Oct 7, 2024
Merged
Show file tree
Hide file tree
Changes from all 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
10 changes: 8 additions & 2 deletions ci/run_integration.sh
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,14 @@ remove_pipenv() {

integration_test_tf() {
echo "Run TF integration test..."
# not using pipenv because we need tensorflow package from the container
python -m pip install -e .[dev]
# since running directly in container, point python to python3.12
ln -sfn /usr/bin/python3.12 /usr/bin/python
ln -sfn /usr/bin/python3.12 /usr/bin/python3
# somehow the base container has blinker which should be removed
apt remove -y python3-blinker python-blinker-doc || true
# pipenv does not work with TensorFlow so using pip
python3.12 -m pip install -e .[dev]
python3.12 -m pip install tensorflow[and-cuda]
export PYTHONPATH=$PWD
testFolder="tests/integration_test"
clean_up_snapshot_and_job
Expand Down
6 changes: 6 additions & 0 deletions setup.cfg
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,10 @@ install_requires =
pyhocon

[options.extras_require]
HE =
tenseal==0.3.15
PSI =
openmined.psi==2.0.5
PT =
torch
torchvision
Expand All @@ -51,6 +55,8 @@ TRACKING =
CONFIG =
omegaconf
app_opt =
%(HE)s
%(PSI)s
%(PT)s
%(SKLEARN)s
%(TRACKING)s
Expand Down
Loading