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.

Sign up for GitHub

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

Jump to bottom

Include apt installation of libgomp1 for docker iamges with lightgbm #2813

Open
wants to merge 5 commits into
base: develop
Choose a base branch
from
+7 −0
Open

Include apt installation of libgomp1 for docker iamges with lightgbm #2813

Show file tree
Hide file tree
Changes from 2 commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
867032b
Include apt installation of libgomp1 for docker iamges with lightgbm
AlexejPenner Jun 28, 2024
c593a16
inlcude apt packages from required integrations
wjayesh Jul 3, 2024
db8dac3
Merge branch 'develop' into bugfix/grow-10-lightgbm-apt-requirements
AlexejPenner Jul 8, 2024
4df46a9
Reformatted
AlexejPenner Jul 8, 2024
5f9ce50
Merge branch 'develop' into bugfix/grow-10-lightgbm-apt-requirements
AlexejPenner Jul 8, 2024
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
1 change: 1 addition & 0 deletions src/zenml/integrations/lightgbm/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ class LightGBMIntegration(Integration):

NAME = LIGHTGBM
REQUIREMENTS = ["lightgbm>=1.0.0"]
APT_PACKAGES = ["libgomp1"]

@classmethod
def activate(cls) -> None:
Expand Down
6 changes: 6 additions & 0 deletions src/zenml/utils/pipeline_docker_image_builder.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -290,6 +290,12 @@ def build_docker_image(
apt_packages = docker_settings.apt_packages.copy()
if docker_settings.install_stack_requirements:
apt_packages += stack.apt_packages

# include apt packages from all required integrations
for integration in docker_settings.required_integrations:
# get the integration
integration = integration_registry.integrations[integration]
apt_packages += integration.APT_PACKAGES

if apt_packages:
logger.info(
Expand Down
Loading