def model_type()

in src/neo_loader/helpers/tf_model_helper.py [0:0]

• 7 lines of code
• 5 McCabe index (conditional complexity)

    def model_type(self) -> bool:
        if self.model_path.is_file() and self.model_path.suffix in ['.pb', '.pbtxt']:
            return TFModelFormat.FrozenGraphModel
        elif self.model_path.is_dir() and self.model_path.joinpath('variables').exists():
            return TFModelFormat.SavedModel
        else:
            raise Exception(f"Encountered invalid model file format. {self.model_path.as_posix}")