Skip to content

Commit 1bb6119

Browse files
fix: fix recommender install (#660)
Signed-off-by: yashasvi <yashasvi@ibm.com>
1 parent 3f08887 commit 1bb6119

2 files changed

Lines changed: 2 additions & 3 deletions

File tree

build/nvcr.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ RUN if [[ "${ENABLE_SCANNER}" == "true" ]]; then \
7878
pip install --no-cache-dir ${SOURCE_DIR}[scanner-dev]; \
7979
fi
8080
RUN if [[ "${ENABLE_RECOMMENDER}" == "true" ]]; then \
81-
python -m pip install --user "$(head bdist_name)[tuning-config-recommender]"; \
81+
pip install --no-cache-dir ${SOURCE_DIR}[tuning_config_recommender]; \
8282
fi
8383

8484
# cleanup

pyproject.toml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@ dependencies = [
4040
"simpleeval>=0.9.13,<2.0",
4141
"pillow>=12.1.1",
4242
"kernels<=0.9.0",
43-
"tuning_config_recommender>=0.1.7",
4443
]
4544

4645
[project.optional-dependencies]
@@ -61,7 +60,7 @@ fms-accel-all = [
6160
"fms-acceleration-moe",
6261
"fms-acceleration-odm"
6362
]
64-
tuning-config-recommender=["tuning-config-recommender>=0.1.5"]
63+
tuning_config_recommender=["tuning_config_recommender>=0.1.7"]
6564

6665
[tool.setuptools.packages.find]
6766
exclude = ["tests", "tests.*"]

0 commit comments

Comments
 (0)