Add __version__ property. (#543)

Author: AdamHillier

larq_compute_engine/__init__.py

@@ -1,4 +1,12 @@
 from larq_compute_engine.mlir.python.converter import convert_keras_model
 from larq_compute_engine.tflite.python import interpreter as testing
 
+try:
+    from importlib import metadata  # type: ignore
+except ImportError:
+    # Running on pre-3.8 Python; use importlib-metadata package
+    import importlib_metadata as metadata  # type: ignore
+
+__version__ = metadata.version("larq_compute_engine")
+
 __all__ = ["convert_keras_model", "testing"]

larq_compute_engine/mlir/python/converter_test.py

@@ -5,6 +5,8 @@
 import larq_zoo as lqz
 from tensorflow.python.eager import context
 
+sys.modules["importlib.metadata"] = mock.MagicMock()
+sys.modules["importlib_metadata"] = mock.MagicMock()
 sys.modules["larq_compute_engine.mlir._graphdef_tfl_flatbuffer"] = mock.MagicMock()
 sys.modules[
     "larq_compute_engine.tflite.python.interpreter_wrapper_lite"

setup.py

@@ -30,7 +30,12 @@ def has_ext_modules(self):
     packages=find_packages(),
     ext_modules=ext_modules,
     url="https://larq.dev/",
-    install_requires=["flatbuffers>=1.12", "packaging>=19", "tqdm>=4"],
+    install_requires=[
+        "flatbuffers>=1.12",
+        "packaging>=19",
+        "tqdm>=4",
+        "importlib-metadata ~= 2.0 ; python_version<'3.8'",
+    ],
     extras_require={
         "tensorflow": ["tensorflow>=1.14"],
         "tensorflow_gpu": ["tensorflow-gpu>=1.14"],