remove old context attribute from backends

tum-ei-eda · Jun 27, 2022 · c43cd7e · c43cd7e
1 parent adc89db
commit c43cd7e
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 10 deletions.
diff --git a/mlonmcu/flow/backend.py b/mlonmcu/flow/backend.py
@@ -44,13 +44,11 @@ def __init__(
         framework="",
         features=None,
         config=None,
-        context=None,
     ):
         self.framework = framework
         self.config = config if config else {}
         self.features = self.process_features(features)
         self.config = filter_config(self.config, self.name, self.DEFAULTS, self.REQUIRED)
-        self.context = context
         self.artifacts = []
         self.supported_fmts = []
         self.tuner = None

diff --git a/mlonmcu/flow/tflm/backend/backend.py b/mlonmcu/flow/tflm/backend/backend.py
@@ -31,8 +31,8 @@ class TFLMBackend(Backend):
 
     REQUIRED = []
 
-    def __init__(self, features=None, config=None, context=None):
-        super().__init__(framework="tflm", config=config, context=context)
+    def __init__(self, features=None, config=None):
+        super().__init__(framework="tflm", config=config)
         self.model = None
         self.supported_formats = [ModelFormats.TFLITE]
 

diff --git a/mlonmcu/flow/tflm/backend/tflmc.py b/mlonmcu/flow/tflm/backend/tflmc.py
@@ -44,8 +44,8 @@ class TFLMCBackend(TFLMBackend):
 
     REQUIRED = TFLMBackend.REQUIRED + ["tflmc.exe"]
 
-    def __init__(self, features=None, config=None, context=None):
-        super().__init__(features=features, config=config, context=context)
+    def __init__(self, features=None, config=None):
+        super().__init__(features=features, config=config)
         self.model_data = None
         self.prefix = "model"  # Without the _
         self.artifacts = (

diff --git a/mlonmcu/flow/tflm/backend/tflmi.py b/mlonmcu/flow/tflm/backend/tflmi.py
@@ -325,8 +325,8 @@ class TFLMIBackend(TFLMBackend):
 
     REQUIRED = TFLMBackend.REQUIRED + []
 
-    def __init__(self, features=None, config=None, context=None):
-        super().__init__(features=features, config=config, context=context)
+    def __init__(self, features=None, config=None):
+        super().__init__(features=features, config=config)
         self.codegen = TFLMICodegen()
         self.model_data = None
         self.prefix = "model"  # Without the _

diff --git a/mlonmcu/flow/tvm/backend/backend.py b/mlonmcu/flow/tvm/backend/backend.py
@@ -63,8 +63,8 @@ class TVMBackend(Backend):
 
     REQUIRED = ["tvm.build_dir", "tvm.pythonpath", "tvm.configs_dir"]
 
-    def __init__(self, features=None, config=None, context=None):
-        super().__init__(framework="tvm", features=features, config=config, context=context)
+    def __init__(self, features=None, config=None):
+        super().__init__(framework="tvm", features=features, config=config)
 
         self.model = None  # Actual filename!
         self.model_info = None