We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ebb54cb commit a0d2ebcCopy full SHA for a0d2ebc
keras_remote/backend/execution.py
@@ -218,9 +218,11 @@ def _build_container(ctx: JobContext) -> None:
218
ctx.image_uri = ctx.container_image
219
logging.info("Using custom container: %s", ctx.image_uri)
220
else:
221
+ import sys
222
logging.info("Building container image...")
223
+ py_version = f"{sys.version_info.major}.{sys.version_info.minor}"
224
ctx.image_uri = container_builder.get_or_build_container(
- base_image="python:3.12-slim",
225
+ base_image=f"python:{py_version}-slim",
226
requirements_path=ctx.requirements_path,
227
accelerator_type=ctx.accelerator,
228
project=ctx.project,
0 commit comments