def __extract_model_data_types_and_metadata_from_model_file()

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

• 18 lines of code
• 4 McCabe index (conditional complexity)

    def __extract_model_data_types_and_metadata_from_model_file(self) -> None:
        tflite_model_helper = TFLiteModelHelper(self.__model_file.as_posix())
        try:
            tflite_model_helper.load_model()
        except Exception as ex:
            logger.exception("Failed to load TFLite model.%s" % repr(ex))
            raise RuntimeError("InputConfiguration: Framework cannot load TFLite model: {}".format(ex))
        else:
            try:
                tflite_model_helper.extract_input_and_output_tensors(user_shape_dict=self.data_shape)
                self.__model = tflite_model_helper.tflite_model
                self.__data_types = tflite_model_helper.input_dtypes_dict
                self._metadata = tflite_model_helper.get_metadata()
            except ImportError:
                raise
            except Exception as ex:
                logging.exception("Unable to infer tensor data type for all inputs/outputs %s" % repr(ex))
                raise RuntimeError("InputConfiguration: Framework cannot load TFLite model. Unable to infer tensor data type for all inputs/outputs: {}".format(ex))