Skip to content

Commit

Permalink
fix pip cache, pip-compile lock reqs
Browse files Browse the repository at this point in the history
Signed-off-by: Jack Luar <[email protected]>
  • Loading branch information
luarss committed Oct 31, 2024
1 parent 38b66e0 commit d947e62
Show file tree
Hide file tree
Showing 5 changed files with 463 additions and 9 deletions.
1 change: 1 addition & 0 deletions tools/AutoTuner/.gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -10,3 +10,4 @@ __pycache__/
# Autotuner env
autotuner_env
.env
.venv
15 changes: 15 additions & 0 deletions tools/AutoTuner/Makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
clearcache:
@echo "Cleaning python cache"
@find . -type d -name __pycache__ -exec rm -r {} \+

init:
@echo "Installing python environment"
@./installer.sh

reqs:
@echo "Compiling requirements"
@rm -f requirements.txt
@.venv/bin/pip-compile --output-file=requirements.txt requirements.in

test:
@echo "Running tests"
3 changes: 2 additions & 1 deletion tools/AutoTuner/installer.sh
Original file line number Diff line number Diff line change
Expand Up @@ -7,5 +7,6 @@ script_dir="$(dirname "${BASH_SOURCE[0]}")"
venv_name="autotuner_env"
python3 -m venv "$script_dir/$venv_name"
source "$script_dir/$venv_name/bin/activate"
pip3 install -U -r $script_dir/requirements.txt
pip3 cache purge
pip3 install --no-cache-dir -U -r $script_dir/requirements.txt
deactivate
12 changes: 12 additions & 0 deletions tools/AutoTuner/requirements.in
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
ray[default,tune]==2.9.3
ax-platform>=0.3.3,<=0.3.7
hyperopt==0.2.7
optuna==3.6.0
pandas>=2.0,<=2.2.1
bayesian-optimization==1.4.0
colorama==0.4.6
tensorboard>=2.14.0,<=2.16.2
protobuf==3.20.3
SQLAlchemy==1.4.17
urllib3<=1.26.15
pip-tools==7.4.1
Loading

0 comments on commit d947e62

Please sign in to comment.