update

Dockerfile

@@ -10,7 +10,7 @@ WORKDIR /app
 
 # Upgrade pip toolchain and preinstall deps that unblock tf-models-official
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
- && pip install --no-cache-dir "Cython<3" "pyyaml<5.4.0,>=5.1"
+ && pip install --no-cache-dir "Cython<3" 
 
 # Then install your requirements
 COPY requirements.txt /app/requirements.txt

requirements.txt

@@ -6,4 +6,5 @@ numpy==1.24.3
 tensorflow-macos==2.13.1
 tensorflow-metal==1.0.0    # optional but speeds things up on M1/M2
 tensorflow-privacy==0.8.11
-scikit-learn==1.3.0
+scikit-learn==1.3.0
+pyyaml>=6.0.1