#0: Fixed pre-commit errors (#144)

tenstorrent · Jul 16, 2024 · 9489070 · 9489070
1 parent ecd628c
commit 9489070
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/runtime/tools/python/setup.py b/runtime/tools/python/setup.py
@@ -7,9 +7,9 @@
 from setuptools import setup
 import shutil
 
-TTMLIR_VERSION_MAJOR=os.getenv('TTMLIR_VERSION_MAJOR', '0')
-TTMLIR_VERSION_MINOR=os.getenv('TTMLIR_VERSION_MINOR', '0')
-TTMLIR_VERSION_PATCH=os.getenv('TTMLIR_VERSION_PATCH', '0')
+TTMLIR_VERSION_MAJOR = os.getenv("TTMLIR_VERSION_MAJOR", "0")
+TTMLIR_VERSION_MINOR = os.getenv("TTMLIR_VERSION_MINOR", "0")
+TTMLIR_VERSION_PATCH = os.getenv("TTMLIR_VERSION_PATCH", "0")
 
 __version__ = f"{TTMLIR_VERSION_MAJOR}.{TTMLIR_VERSION_MINOR}.{TTMLIR_VERSION_PATCH}"
 

diff --git a/runtime/tools/python/ttrt/__init__.py b/runtime/tools/python/ttrt/__init__.py
@@ -8,6 +8,7 @@
 
 from pkg_resources import get_distribution
 
+
 def system_desc_as_dict(desc):
     return json.loads(desc.as_json())
 
@@ -17,14 +18,18 @@ def system_desc_as_dict(desc):
 if "TT_METAL_LOGGER_LEVEL" not in os.environ:
     os.environ["TT_METAL_LOGGER_LEVEL"] = "FATAL"
 
+
 def check_version(fb_version):
-    package_name = 'ttrt'
+    package_name = "ttrt"
     try:
         package_version = get_distribution(package_name).version
     except Exception as e:
         print(f"Error retrieving version: {e} for {package_name}")
 
-    assert package_version == fb_version, f"{package_name}=v{package_version} does not match flatbuffer=v{fb_version}"
+    assert (
+        package_version == fb_version
+    ), f"{package_name}=v{package_version} does not match flatbuffer=v{fb_version}"
+
 
 def mlir_sections(fbb):
     d = ttrt.binary.as_dict(fbb)