Merge pull request #288 from kAIto47802/support-python3.12

Add checking for python3.12
optuna · Nov 1, 2024 · 0c6999a · 0c6999a
2 parents d4ed01d + 6ccb8d4
commit 0c6999a
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/tensorflow/tensorflow_estimator_integration.py b/tensorflow/tensorflow_estimator_integration.py
@@ -12,6 +12,7 @@
 """
 
 import shutil
+import sys
 import tempfile
 import urllib
 
@@ -133,4 +134,11 @@ def main():
 
 
 if __name__ == "__main__":
+    if sys.version_info >= (3, 12):
+        raise RuntimeError(
+            "Use `optuna_integration.KerasPruningCallback` instead, "
+            "because TensorFlow recommends migrating from Estimator to Keras APIs. "
+            "For more details, please check the TensorFlow migration guide: "
+            "https://www.tensorflow.org/guide/migrate/migrating_estimator"
+        )
     main()
diff --git a/tensorflow/tensorflow_estimator_simple.py b/tensorflow/tensorflow_estimator_simple.py
@@ -9,6 +9,7 @@
 """
 
 import shutil
+import sys
 import tempfile
 import urllib
 
@@ -123,4 +124,10 @@ def main():
 
 
 if __name__ == "__main__":
+    if sys.version_info >= (3, 12):
+        raise RuntimeError(
+            "Use Keras APIs instead of Estimator API in TensorFlow. "
+            "For more details, please check the TensorFlow migration guide: "
+            "https://www.tensorflow.org/guide/migrate/migrating_estimator"
+        )
     main()