From 6b28bb6e48615c940356e8b709e99d3fd36abd8f Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 30 May 2024 10:37:21 +0200
Subject: [PATCH] Update Exporter for `tf_keras` install (#13231)

---
 ultralytics/engine/exporter.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py
index 0421075cd4..70c0d7a39f 100644
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@@ -819,7 +819,8 @@ class Exporter:
             check_requirements("cmake")  # 'cmake' is needed to build onnxsim on aarch64
         check_requirements(
             (
-                "keras",
+                "keras",  # required by onnx2tf package
+                "tf_keras",  # required by onnx2tf package
                 "onnx>=1.12.0",
                 "onnx2tf>1.17.5,<=1.22.3",
                 "sng4onnx>=1.0.1",