fix typo

Signed-off-by: Robert Steiner <[email protected]>
adap · Nov 4, 2024 · 8b319d4 · 8b319d4
1 parent ba844b0
commit 8b319d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dev/build-docker-image-matrix.py b/dev/build-docker-image-matrix.py
@@ -100,7 +100,7 @@ class CudaVariant:
 ]
 
 
-def remove_path_version(version: str) -> str:
+def remove_patch_version(version: str) -> str:
     return ".".join(version.split(".")[0:2])
 
 
@@ -142,7 +142,7 @@ def remove_path_version(version: str) -> str:
     Variant(
         Distro(DistroName.UBUNTU, ubuntu_version),
         lambda distro, _: f"{DOCKERFILE_ROOT}/base/{distro.name.value}-cuda",
-        lambda distro, flwr_version, python_version, extras: f"{flwr_version}-py{python_version}-cu{remove_path_version(extras.version)}-{distro.name.value}{distro.version}",
+        lambda distro, flwr_version, python_version, extras: f"{flwr_version}-py{python_version}-cu{remove_patch_version(extras.version)}-{distro.name.value}{distro.version}",
         lambda distro, flwr_version, python_version, extras: CUDA_BUILD_ARGS.format(
             python_version=python_version,
             flwr_version=flwr_version,