mttf 1.1.4__py3-none-any.whl → 1.1.6__py3-none-any.whl

mt/keras/__init__.py

@@ -0,0 +1,32 @@
+"""MT package that represents the working Keras 2 from the system."""
+
+from packaging.version import Version
+import tensorflow as tf
+
+tf_ver = Version(tf.__version__)
+if tf_ver >= Version("2.16"):
+    try:
+        import tf_keras
+    except:
+        raise ImportError(
+            f"mt.keras can only work with Keras 2. You have TF version {tf_ver}. Please install tf_keras."
+        )
+    from tf_keras import *
+
+    __version__ = tf_keras.__version__
+    __source__ = "tf_keras"
+else:
+    try:
+        import keras
+
+        kr_ver = Version(keras.__version__)
+    except ImportError:
+        kr_ver = None
+    if kr_ver is None or kr_ver >= Version("3.0"):
+        __version__ = tf.__version__
+        __source__ = "tensorflow.python"
+        from tensorflow.keras import *
+    else:
+        __version__ = keras.__version__
+        __source__ = "keras"
+        from keras import *

mt/tf/init.py

@@ -1,6 +1,6 @@
 """Initialises TensorFlow, monkey-patching if necessary."""
 
-from packaging import version
+from packaging.version import Version
 
 __all__ = ["init"]
 
@@ -11,9 +11,9 @@ def init():
     import tensorflow
     import sys
 
-    tf_ver = version.parse(tensorflow.__version__)
+    tf_ver = Version(tensorflow.__version__)
 
-    if tf_ver < version.parse("2.8"):
+    if tf_ver < Version("2.8"):
         raise ImportError(
             f"The minimum TF version that mttf supports is 2.8. Your TF is {tf_ver}. "
             "Please upgrade."

mt/tf/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 1
-PATCH_VERSION = 4
+PATCH_VERSION = 6
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mttf-1.1.4.dist-info → mttf-1.1.6.dist-info}/METADATA

@@ -1,14 +1,14 @@
 Metadata-Version: 2.4
 Name: mttf
-Version: 1.1.4
-Summary: A package to detect and monkey-patch TensorFlow, for Minh-Tri Pham
+Version: 1.1.6
+Summary: A package to detect and monkey-patch TensorFlow and Keras, for Minh-Tri Pham
 Home-page: https://github.com/inteplus/mttf
 Author: ['Minh-Tri Pham']
 Project-URL: Documentation, https://mtdoc.readthedocs.io/en/latest/mt.tf/mt.tf.html
 Project-URL: Source Code, https://github.com/inteplus/mttf
 License-File: LICENSE
 Requires-Dist: pyyaml
-Requires-Dist: mtbase>=4.32
+Requires-Dist: mtbase>=4.32.3
 Requires-Dist: mtnet>=0.3.2
 Dynamic: author
 Dynamic: home-page

{mttf-1.1.4.dist-info → mttf-1.1.6.dist-info}/RECORD

@@ -1,8 +1,9 @@
+mt/keras/__init__.py,sha256=Qv3x6wnkG-ytMPzgQ8xPNe4NxxiD4z9RG23rvvLyRvg,883
 mt/tf/__init__.py,sha256=M8xiJNdrAUJZgiZTOQOdfkehjO-CYzGpoxh5HVGBkms,338
-mt/tf/init.py,sha256=Hbp0-daCDTUEkQiDwNpbv2qqP7tr9qCyBFUafMklkos,1298
+mt/tf/init.py,sha256=BUpw33uyA_DmeJjrN2fX4MIs8MynKxkwgc2oTGTqssU,1294
 mt/tf/mttf_version.py,sha256=ha53i-H9pE-crufFttUECgXHwPvam07zMKzApUts1Gs,206
 mt/tf/utils.py,sha256=Copl5VM0PpuFUchK-AcBuGO6QitDwHcEs4FruZb2GAI,2460
-mt/tf/version.py,sha256=wwYizgo5s_gw0_URJ5ObnIKjwfoG5wdSZG30y4tXWK4,206
+mt/tf/version.py,sha256=WILNiQZrHE_E8TaKdRSqSywIA2x-8Z8oGUEwvLmUqMI,206
 mt/tf/keras_applications/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mt/tf/keras_applications/mobilenet_v3_split.py,sha256=1oPB3EX3k3c7iju9Ksuw9xyv32-mOKPs3uy2Mk5tLd8,19716
 mt/tf/keras_applications/mobilevit.py,sha256=VsKB_U-f9jwUEjpd0eq-YXL4rDsuAbKQ0yIzkbMfLzw,9949
@@ -19,14 +20,14 @@ mt/tfc/__init__.py,sha256=XFXAbmEi5uTAuZj797cBqw9ZBnVy6ptxP0TTzVauMNk,8446
 mt/tfg/__init__.py,sha256=6Ly2QImAyQTsg_ZszuAuK_L2n56v89Cix9yYmMVk0CM,304
 mt/tfp/__init__.py,sha256=AQkGCkmDRwswEt3qoOSpxe-fZekx78sHHBs2ZVz33gc,383
 mt/tfp/real_nvp.py,sha256=U9EmkXGqFcvtS2yeh5_RgbKlVKKlGFGklAb7Voyazz4,4440
-mttf-1.1.4.data/scripts/dmt_pipi.sh,sha256=NNsj4P332unHMqU4mAFjU9PQvxQ8TK5XQ42LC29IZY8,510
-mttf-1.1.4.data/scripts/dmt_twineu.sh,sha256=KZhcYwuCW0c36tWcOgCe7uxJmS08rz-J6YNY76Exy4M,193
-mttf-1.1.4.data/scripts/pipi.sh,sha256=kdo96bdaKq2QIa52Z4XFSiGPcbDm09SAU9cju6I2Lxo,289
-mttf-1.1.4.data/scripts/wml_nexus.py,sha256=kW0ju8_kdXc4jOjhdzKiMsFuO1MNpHmu87skrhu9SEg,1492
-mttf-1.1.4.data/scripts/wml_pipi.sh,sha256=CuidIcbuxyXSBNQqYRhCcSC8QbBaSGnQX0KAIFaIvKA,499
-mttf-1.1.4.data/scripts/wml_twineu.sh,sha256=av1JLN765oOWC5LPkv2eSWIVof26y60601tMGkuYdb8,180
-mttf-1.1.4.dist-info/licenses/LICENSE,sha256=e_JtcszdGZ2ZGfjcymTGrcxFj_9XPicZOVtnsrPvruk,1070
-mttf-1.1.4.dist-info/METADATA,sha256=yr74gOSqTLrx-cOm9pZPjFFGk5rDR6npCbs9Oy6pJV0,555
-mttf-1.1.4.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-mttf-1.1.4.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
-mttf-1.1.4.dist-info/RECORD,,
+mttf-1.1.6.data/scripts/dmt_pipi.sh,sha256=NNsj4P332unHMqU4mAFjU9PQvxQ8TK5XQ42LC29IZY8,510
+mttf-1.1.6.data/scripts/dmt_twineu.sh,sha256=KZhcYwuCW0c36tWcOgCe7uxJmS08rz-J6YNY76Exy4M,193
+mttf-1.1.6.data/scripts/pipi.sh,sha256=kdo96bdaKq2QIa52Z4XFSiGPcbDm09SAU9cju6I2Lxo,289
+mttf-1.1.6.data/scripts/wml_nexus.py,sha256=kW0ju8_kdXc4jOjhdzKiMsFuO1MNpHmu87skrhu9SEg,1492
+mttf-1.1.6.data/scripts/wml_pipi.sh,sha256=CuidIcbuxyXSBNQqYRhCcSC8QbBaSGnQX0KAIFaIvKA,499
+mttf-1.1.6.data/scripts/wml_twineu.sh,sha256=av1JLN765oOWC5LPkv2eSWIVof26y60601tMGkuYdb8,180
+mttf-1.1.6.dist-info/licenses/LICENSE,sha256=e_JtcszdGZ2ZGfjcymTGrcxFj_9XPicZOVtnsrPvruk,1070
+mttf-1.1.6.dist-info/METADATA,sha256=O8Ly1k7_qsC-jBLBywSh3jwYFRTCuaRfmfd3ovnAUos,567
+mttf-1.1.6.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
+mttf-1.1.6.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
+mttf-1.1.6.dist-info/RECORD,,

{mttf-1.1.4.dist-info → mttf-1.1.6.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.0)
+Generator: setuptools (80.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any