fix health check dep

Dockerfile

@@ -2,6 +2,8 @@
 ARG BASE_IMG=python:3.10-slim
 FROM $BASE_IMG
 
+ARG GRPC_HEALTH_PROBE_VERSION=""
+
 # Requirements (use MNIST Keras as default)
 ARG REQUIREMENTS=""
 
@@ -13,12 +15,17 @@ COPY config/settings-reducer.yaml.template /app/config/settings-reducer.yaml
 COPY $REQUIREMENTS /app/config/requirements.txt
 
 # Install developer tools (needed for psutil)
-RUN apt-get update && apt-get install -y python3-dev gcc wget
+RUN apt-get update && apt-get install -y python3-dev gcc
 
 # Install grpc health probe checker
-RUN GRPC_HEALTH_PROBE_VERSION=v0.4.24 && \
+RUN if [ ! -z "$GRPC_HEALTH_PROBE_VERSION" ]; then \
+  apt-get install -y wget && \
   wget -qO/bin/grpc_health_probe https://github.com/grpc-ecosystem/grpc-health-probe/releases/download/${GRPC_HEALTH_PROBE_VERSION}/grpc_health_probe-linux-amd64 && \
-  chmod +x /bin/grpc_health_probe
+  chmod +x /bin/grpc_health_probe && \
+  apt-get remove -y wget && apt autoremove -y; \
+  else \
+  echo "No grpc_health_probe version specified, skipping installation"; \
+  fi
 
 
 # Create FEDn app directory

docker-compose.yaml

@@ -91,6 +91,7 @@ services:
       context: .
       args:
         BASE_IMG: ${BASE_IMG:-python:3.10-slim}
+        GRPC_HEALTH_PROBE_VERSION: v0.4.24
     working_dir: /app
     volumes:
       - ${HOST_REPO_DIR:-.}/fedn:/app/fedn