Skip to content

Commit 3ba390a

Browse files
authored
Merge branch 'main' into yash-tfv5
Signed-off-by: Yash Mehan <71321431+yash4242@users.noreply.github.com>
2 parents d5e1862 + 1bb6119 commit 3ba390a

2 files changed

Lines changed: 2 additions & 2 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 & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ fms-accel-all = [
6161
"fms-acceleration-moe",
6262
"fms-acceleration-odm"
6363
]
64-
tuning-config-recommender=["tuning-config-recommender>=0.1.5"]
64+
tuning_config_recommender=["tuning_config_recommender>=0.1.7"]
6565

6666
[tool.setuptools.packages.find]
6767
exclude = ["tests", "tests.*"]

0 commit comments

Comments
 (0)