From a73aa17a106dab2ed21f2573bb15d4326c79adbb Mon Sep 17 00:00:00 2001
From: Francisco Rossi <frrossi@fi.uba.ar>
Date: Sun, 23 Nov 2025 13:24:46 -0300
Subject: [PATCH] Pin tf_keras version to 2.17

---
 docker/Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 4876623..d7f97d3 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -9,6 +9,7 @@ RUN pip3 install \
 
 # Set backward compatibility for tfmot
 RUN pip3 install tf_keras --no-deps
+RUN pip3 install "tf_keras~=2.17.0" --no-deps
 ENV TF_USE_LEGACY_KERAS=1
 
 # Install linting tools