def load_model()

in src/neo_loader/tflite_model_loader.py [0:0]

• 17 lines of code
• 3 McCabe index (conditional complexity)

    def load_model(self) -> None:
        self.__extract_model_file_from_model_artifacts()
        self.__extract_model_data_types_and_metadata_from_model_file()
        try:
            self._relay_module_object, self._params = relay.frontend.from_tflite(
                self.__model,
                shape_dict=self.data_shape,
                dtype_dict=self.__data_types
            )
            self._relay_module_object = self.convert_layout(self._relay_module_object)
            self.update_missing_metadata()
        except OpError:
            raise
        except Exception as e:
            logger.exception("Failed to convert tensorflow model. %s" % repr(e))
            msg = "InputConfiguration: TVM cannot convert TFLite model. Please make sure the framework you selected is correct. {}".format(e)
            raise RuntimeError(msg)