diff options
Diffstat (limited to 'src/mlia/target/cortex_a/config.py')
| -rw-r--r-- | src/mlia/target/cortex_a/config.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/mlia/target/cortex_a/config.py b/src/mlia/target/cortex_a/config.py index f91031e..4f33f3d 100644 --- a/src/mlia/target/cortex_a/config.py +++ b/src/mlia/target/cortex_a/config.py @@ -15,12 +15,12 @@ class CortexAConfiguration(TargetProfile): def __init__(self, **kwargs: Any) -> None: """Init Cortex-A target configuration.""" target = kwargs["target"] - super().__init__(target) + backend_config = kwargs.get("backend") + super().__init__(target, backend_config) - self.backend_config = kwargs.get("backend") - self.armnn_tflite_delegate_version = kwargs["backend"]["armnn-tflite-delegate"][ - "version" - ] + self.armnn_tflite_delegate_version = self.backend_config[ + "armnn-tflite-delegate" + ]["version"] def verify(self) -> None: """Check the parameters.""" |