def __extract_tf_graph()

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

• 19 lines of code
• 6 McCabe index (conditional complexity)

    def __extract_tf_graph(self):
        if self.__is_tf2_model:
            try:
                logger.info("Loading TF model for potential TF 2.x model.")
                self.__tf_graph = self.__tf_model_helper.get_tf_graph_from_graph_model_v2()
            except Exception as e:
                logger.exception("Failed to load TF model. %s" % repr(e))
                raise RuntimeError("InputConfiguration: Framework cannot load Tensorflow model: {}".format(e))
        else:
            try:
                logger.info("Loading TF model from TFParser.")
                self.__tf_graph = TFParser(self.__model_path.as_posix(), self.__output_tensor_names).parse()
            except Exception as e:
                # Temp workaround for TF2 models, remove the logic when TF2 is introduced
                try:
                    logger.warning("Failed to load TF model from TFParser, will try to load with compat.v2. %s" % repr(e))
                    self.__tf_graph = self.__tf_model_helper.get_tf_graph_from_graph_model_v2()
                except Exception as error:
                    logger.exception("Failed to load TF model. %s" % repr(e))
                    raise RuntimeError("InputConfiguration: Framework cannot load Tensorflow model: {}".format(e))