ENGPLAT-106: Upgrade python from 3.7 -> 3.9

GitOrigin-RevId: f57283a129519c016dcc1ec20442265a57b62fae
gretelai · May 17, 2022 · 82317b0 · 82317b0
1 parent 33d76f1
commit 82317b0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 9 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,9 @@
-loky==2.8.0
+loky==2.9.0
 numpy>=1.18.0
 pandas>=1.1.0
-sentencepiece==0.1.91
-smart_open>=2.1.0,<3.0
-tensorflow==2.4.0
-tensorflow_privacy==0.5.1
+sentencepiece==0.1.96
+smart_open>=2.1.0,<6.0
+tensorflow==2.8.0
+tensorflow_privacy==0.7.3
 torch==1.11.0
 tqdm<5.0
diff --git a/src/gretel_synthetics/tensorflow/dp_model.py b/src/gretel_synthetics/tensorflow/dp_model.py
@@ -41,9 +41,7 @@ def build_dp_model(store, batch_size, vocab_size) -> tf.keras.Sequential:
     logging.warning("Experimental: Differentially private training enabled")
 
     try:
-        recurrent_v2 = importlib.import_module(
-            "tensorflow.python.keras.layers.recurrent_v2"
-        )
+        recurrent_v2 = importlib.import_module("keras.layers.recurrent_v2")
         # NOTE: This patches the LSTMs to use the new Keras 2.4.x code paths
         # and will have no effect when the module function is removed
         use_new_code = getattr(recurrent_v2, "_use_new_code", None)

diff --git a/utils-requirements.txt b/utils-requirements.txt
@@ -1,4 +1,4 @@
 category-encoders==2.2.2
-dython==0.5.1
+dython==0.7.0
 scikit-learn>=0.24
 scipy>=1.4.1