# -*- coding: utf-8 -*-
# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: megamolbart.proto
"""Generated protocol buffer code."""
from google.protobuf import descriptor as _descriptor
from google.protobuf import descriptor_pool as _descriptor_pool
from google.protobuf import message as _message
from google.protobuf import reflection as _reflection
from google.protobuf import symbol_database as _symbol_database
# @@protoc_insertion_point(imports)

_sym_db = _symbol_database.Default()




DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x11megamolbart.proto\x12!nemo_chem.models.megamolbart.grpc\"L\n\tInputSpec\x12\x0c\n\x04smis\x18\x01 \x03(\t\x12\x15\n\rhidden_states\x18\x02 \x03(\x02\x12\x0b\n\x03\x64im\x18\x03 \x03(\x05\x12\r\n\x05masks\x18\x04 \x03(\x08\"a\n\nOutputSpec\x12\x0c\n\x04smis\x18\x01 \x03(\t\x12\x15\n\rhidden_states\x18\x02 \x03(\x02\x12\x12\n\nembeddings\x18\x03 \x03(\x02\x12\x0b\n\x03\x64im\x18\x04 \x03(\x05\x12\r\n\x05masks\x18\x05 \x03(\x08\x32\xe7\x02\n\x11GenerativeSampler\x12r\n\x11SmilesToEmbedding\x12,.nemo_chem.models.megamolbart.grpc.InputSpec\x1a-.nemo_chem.models.megamolbart.grpc.OutputSpec\"\x00\x12o\n\x0eSmilesToHidden\x12,.nemo_chem.models.megamolbart.grpc.InputSpec\x1a-.nemo_chem.models.megamolbart.grpc.OutputSpec\"\x00\x12m\n\x0cHiddenToSmis\x12,.nemo_chem.models.megamolbart.grpc.InputSpec\x1a-.nemo_chem.models.megamolbart.grpc.OutputSpec\"\x00\x62\x06proto3')



_INPUTSPEC = DESCRIPTOR.message_types_by_name['InputSpec']
_OUTPUTSPEC = DESCRIPTOR.message_types_by_name['OutputSpec']
InputSpec = _reflection.GeneratedProtocolMessageType('InputSpec', (_message.Message,), {
  'DESCRIPTOR' : _INPUTSPEC,
  '__module__' : 'megamolbart_pb2'
  # @@protoc_insertion_point(class_scope:nemo_chem.models.megamolbart.grpc.InputSpec)
  })
_sym_db.RegisterMessage(InputSpec)

OutputSpec = _reflection.GeneratedProtocolMessageType('OutputSpec', (_message.Message,), {
  'DESCRIPTOR' : _OUTPUTSPEC,
  '__module__' : 'megamolbart_pb2'
  # @@protoc_insertion_point(class_scope:nemo_chem.models.megamolbart.grpc.OutputSpec)
  })
_sym_db.RegisterMessage(OutputSpec)

_GENERATIVESAMPLER = DESCRIPTOR.services_by_name['GenerativeSampler']
if _descriptor._USE_C_DESCRIPTORS == False:

  DESCRIPTOR._options = None
  _INPUTSPEC._serialized_start=56
  _INPUTSPEC._serialized_end=132
  _OUTPUTSPEC._serialized_start=134
  _OUTPUTSPEC._serialized_end=231
  _GENERATIVESAMPLER._serialized_start=234
  _GENERATIVESAMPLER._serialized_end=593
# @@protoc_insertion_point(module_scope)
