Skip to content

Commit 1fd6776

Browse files
authored
fix: Move wheels directory if src/wheels exists (#5430)
Signed-off-by: Frost Ming <me@frostming.com>
1 parent e35a602 commit 1fd6776

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/_bentoml_sdk/images.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,8 +247,10 @@ def _freeze_python_requirements(
247247
req.name and req.name.lower() == "bentoml" and req.link is not None
248248
for req in requirements_file.requirements
249249
)
250+
src_wheels = bento_fs / "src/wheels"
250251
wheels_folder = py_folder.joinpath("wheels")
251-
wheels_folder.mkdir(parents=True, exist_ok=True)
252+
if src_wheels.exists():
253+
shutil.move(src_wheels, wheels_folder)
252254

253255
with requirements_in.open("w") as f:
254256
f.write(requirements_file.dumps(preserve_one_empty_line=True))

0 commit comments

Comments
 (0)