lint

tum-ei-eda · Apr 11, 2024 · 5f198eb · 5f198eb
1 parent 607c5c6
commit 5f198eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/mlonmcu/models/frontend.py b/mlonmcu/models/frontend.py
@@ -229,11 +229,11 @@ def generate_input_data(self, input_names, input_types, input_shapes, input_quan
                                 if UPPER is None:
                                     UPPER = dtype_info.max
                                 else:
-                                    assert UPPER <= dtype_info.max, f"Out of dtype bound"
+                                    assert UPPER <= dtype_info.max, "Out of dtype bound"
                                 if LOWER is None:
                                     LOWER = dtype_info.min
                                 else:
-                                    assert LOWER >= dtype_info.min, f"Out of dtype bound"
+                                    assert LOWER >= dtype_info.min, "Out of dtype bound"
                             else:
                                 raise RuntimeError(f"Unsupported dtype: {gen_dtype}")
                             RANGE = UPPER - LOWER
@@ -244,7 +244,8 @@ def generate_input_data(self, input_names, input_types, input_shapes, input_quan
                             # if "float" in dtype:
                             #     arr = np.random.rand(*shape).astype(dtype)
                             # elif "int" in dtype:
-                            #     arr = np.random.randint(np.iinfo(dtype).min, np.iinfo(dtype).max, size=shape, dtype=dtype)
+                            #     arr = np.random.randint(np.iinfo(dtype).min,
+                            #     np.iinfo(dtype).max, size=shape, dtype=dtype)
                             # else:
                             #     assert False
                         # Quantize if required

diff --git a/mlonmcu/platform/tvm/tvm_target_platform.py b/mlonmcu/platform/tvm/tvm_target_platform.py
@@ -18,14 +18,12 @@
 #
 """TVM Target Platform"""
 import os
-from pathlib import Path
 from mlonmcu.config import str2bool
 from .tvm_rpc_platform import TvmRpcPlatform
 from ..platform import TargetPlatform
 from mlonmcu.target import get_targets
 from mlonmcu.target.target import Target
 from .tvm_target import create_tvm_platform_target
-from mlonmcu.artifact import Artifact, ArtifactFormat
 from mlonmcu.flow.tvm.backend.tvmc_utils import (
     get_bench_tvmc_args,
     get_data_tvmc_args,