test

mindee · Apr 24, 2024 · 0cab9d0 · 0cab9d0
1 parent 06de30d
commit 0cab9d0
Show file tree

Hide file tree

Showing 10 changed files with 47 additions and 5 deletions.
diff --git a/doctr/file_utils.py b/doctr/file_utils.py
@@ -85,6 +85,11 @@
     )
 
 
+def ensure_keras_v2() -> None:  # pragma: no cover
+    if not os.environ.get("TF_USE_LEGACY_KERAS") == "1":
+        os.environ["TF_USE_LEGACY_KERAS"] = "1"
+
+
 def requires_package(name: str, extra_message: Optional[str] = None) -> None:  # pragma: no cover
     """
     package requirement helper

diff --git a/references/classification/latency_tensorflow.py b/references/classification/latency_tensorflow.py
@@ -9,14 +9,18 @@
 import os
 import time
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 
-from doctr.models import classification  # isort: skip
-
 import numpy as np
 import tensorflow as tf
 
+from doctr.models import classification
+
 
 def main(args):
     if args.gpu:

diff --git a/references/classification/train_tensorflow_character.py b/references/classification/train_tensorflow_character.py
@@ -5,6 +5,10 @@
 
 import os
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 

diff --git a/references/classification/train_tensorflow_orientation.py b/references/classification/train_tensorflow_orientation.py
@@ -5,6 +5,10 @@
 
 import os
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 

diff --git a/references/detection/evaluate_tensorflow.py b/references/detection/evaluate_tensorflow.py
@@ -5,6 +5,10 @@
 
 import os
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 from doctr.file_utils import CLASS_NAME
 
 os.environ["USE_TF"] = "1"

diff --git a/references/detection/latency_tensorflow.py b/references/detection/latency_tensorflow.py
@@ -9,14 +9,19 @@
 import os
 import time
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 
-from doctr.models import detection  # isort: skip
 
 import numpy as np
 import tensorflow as tf
 
+from doctr.models import detection
+
 
 def main(args):
     if args.gpu:

diff --git a/references/detection/train_tensorflow.py b/references/detection/train_tensorflow.py
@@ -5,6 +5,10 @@
 
 import os
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 

diff --git a/references/recognition/evaluate_tensorflow.py b/references/recognition/evaluate_tensorflow.py
@@ -5,6 +5,10 @@
 
 import os
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 

diff --git a/references/recognition/latency_tensorflow.py b/references/recognition/latency_tensorflow.py
@@ -9,14 +9,18 @@
 import os
 import time
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 
-from doctr.models import recognition  # isort: skip
-
 import numpy as np
 import tensorflow as tf
 
+from doctr.models import recognition
+
 
 def main(args):
     if args.gpu:

diff --git a/references/recognition/train_tensorflow.py b/references/recognition/train_tensorflow.py
@@ -5,6 +5,10 @@
 
 import os
 
+from doctr.file_utils import ensure_keras_v2
+
+ensure_keras_v2()
+
 os.environ["USE_TF"] = "1"
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"