hooks: update tensorflow hook for tensorflow v2.14.0

Add `ml_dtypes` to hidden imports.
rokm · Sep 29, 2023 · 57299b1 · 57299b1
1 parent 39b588b
commit 57299b1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/news/647.update.rst b/news/647.update.rst
@@ -0,0 +1 @@
+Update ``tensorflow`` hook for compatibility with ``tensorflow`` 2.14.0.
diff --git a/requirements-test-libraries.txt b/requirements-test-libraries.txt
@@ -117,7 +117,7 @@ limits==3.6.0
 # great-expectations as of v0.16.10 does not officially support python 3.11 yet, and the pyspark
 # compatibility wrapper seems to be broken for python 3.11 in this version.
 great-expectations==0.17.19; python_version < '3.11'
-tensorflow==2.13.0
+tensorflow==2.14.0; python_version >= '3.9'
 pyshark==0.6.0
 opencv-python==4.8.0.76
 hydra-core==1.3.2

diff --git a/src/_pyinstaller_hooks_contrib/hooks/stdhooks/hook-tensorflow.py b/src/_pyinstaller_hooks_contrib/hooks/stdhooks/hook-tensorflow.py
@@ -72,6 +72,10 @@ def _submodules_filter(x):
     if is_module_satisfies("tensorflow >= 2.6.0"):
         hiddenimports += collect_submodules('keras')
 
+    # Starting with 2.14.0, we need `ml_dtypes` among hidden imports.
+    if is_module_satisfies("tensorflow >= 2.14.0"):
+        hiddenimports += ['ml_dtypes']
+
 excludedimports = excluded_submodules
 
 # Suppress warnings for missing hidden imports generated by this hook.