don't use legacy keras anymore

Author: rvankoert

src/main.py

@@ -2,6 +2,7 @@
 import argparse
 # > Standard library
 import os
+os.environ['TF_USE_LEGACY_KERAS'] = '0'
 import time
 import logging
 

tests/test_data_augments.py

@@ -1,6 +1,6 @@
-# Imports
+import os
+os.environ['TF_USE_LEGACY_KERAS'] = '0'
 
-# > Standard library
 import logging
 import unittest
 import sys

tests/test_dataloader.py

@@ -1,6 +1,5 @@
-# Imports
-
-# > Standard library
+import os
+os.environ['TF_USE_LEGACY_KERAS'] = '0'
 import logging
 from pathlib import Path
 import sys

tests/test_datamanager.py

@@ -1,8 +1,6 @@
-# Imports
-import argparse
-# > Standard library
-import logging
 import os
+os.environ['TF_USE_LEGACY_KERAS'] = '0'
+import logging
 from pathlib import Path
 import sys
 import tempfile

tests/test_lr_schedule.py

@@ -1,6 +1,5 @@
-# Imports
-
-# > Standard library
+import os
+os.environ['TF_USE_LEGACY_KERAS'] = '0'
 import logging
 import unittest
 from pathlib import Path

tests/test_model_replacement.py

@@ -1,16 +1,15 @@
-# Imports
+import os
+os.environ['TF_USE_LEGACY_KERAS'] = '0'
+from pathlib import Path
+import sys
+import logging
+import unittest
 
 # > Third party dependencies
 import numpy as np
 import tensorflow as tf
 from tensorflow.keras import layers
 
-# > Standard library
-from pathlib import Path
-import sys
-import logging
-import unittest
-
 
 class TestReplaceLayers(unittest.TestCase):
     """

tests/test_tokenizer.py

@@ -1,8 +1,6 @@
-# Imports
-
-# > Standard library
-import unittest
 import os
+os.environ['TF_USE_LEGACY_KERAS'] = '0'
+import unittest
 import json
 from tempfile import TemporaryDirectory
 import logging