1  
 2  import google.protobuf
 3  from packaging.version import Version
 4  if Version(google.protobuf.__version__).major >= 5:
 5    # -*- coding: utf-8 -*-
 6    # Generated by the protocol buffer compiler.  DO NOT EDIT!
 7    # source: internal.proto
 8    # Protobuf Python Version: 5.26.0
 9    """Generated protocol buffer code."""
10    from google.protobuf import descriptor as _descriptor
11    from google.protobuf import descriptor_pool as _descriptor_pool
12    from google.protobuf import symbol_database as _symbol_database
13    from google.protobuf.internal import builder as _builder
14    # @@protoc_insertion_point(imports)
15  
16    _sym_db = _symbol_database.Default()
17  
18  
19    from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
20  
21  
22    DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0einternal.proto\x12\x0fmlflow.internal\x1a\x15scalapb/scalapb.proto*2\n\x0fInputVertexType\x12\x07\n\x03RUN\x10\x01\x12\x0b\n\x07\x44\x41TASET\x10\x02\x12\t\n\x05MODEL\x10\x03*4\n\x10OutputVertexType\x12\x0e\n\nRUN_OUTPUT\x10\x01\x12\x10\n\x0cMODEL_OUTPUT\x10\x02\x42#\n\x19org.mlflow.internal.proto\x90\x01\x01\xe2?\x02\x10\x01')
23  
24    _globals = globals()
25    _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
26    _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'internal_pb2', _globals)
27    if not _descriptor._USE_C_DESCRIPTORS:
28      _globals['DESCRIPTOR']._loaded_options = None
29      _globals['DESCRIPTOR']._serialized_options = b'\n\031org.mlflow.internal.proto\220\001\001\342?\002\020\001'
30      _globals['_INPUTVERTEXTYPE']._serialized_start=58
31      _globals['_INPUTVERTEXTYPE']._serialized_end=108
32      _globals['_OUTPUTVERTEXTYPE']._serialized_start=110
33      _globals['_OUTPUTVERTEXTYPE']._serialized_end=162
34    # @@protoc_insertion_point(module_scope)
35  
36  else:
37    # -*- coding: utf-8 -*-
38    # Generated by the protocol buffer compiler.  DO NOT EDIT!
39    # source: internal.proto
40    """Generated protocol buffer code."""
41    from google.protobuf.internal import enum_type_wrapper
42    from google.protobuf import descriptor as _descriptor
43    from google.protobuf import descriptor_pool as _descriptor_pool
44    from google.protobuf import message as _message
45    from google.protobuf import reflection as _reflection
46    from google.protobuf import symbol_database as _symbol_database
47    # @@protoc_insertion_point(imports)
48  
49    _sym_db = _symbol_database.Default()
50  
51  
52    from .scalapb import scalapb_pb2 as scalapb_dot_scalapb__pb2
53  
54  
55    DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0einternal.proto\x12\x0fmlflow.internal\x1a\x15scalapb/scalapb.proto*2\n\x0fInputVertexType\x12\x07\n\x03RUN\x10\x01\x12\x0b\n\x07\x44\x41TASET\x10\x02\x12\t\n\x05MODEL\x10\x03*4\n\x10OutputVertexType\x12\x0e\n\nRUN_OUTPUT\x10\x01\x12\x10\n\x0cMODEL_OUTPUT\x10\x02\x42#\n\x19org.mlflow.internal.proto\x90\x01\x01\xe2?\x02\x10\x01')
56  
57    _INPUTVERTEXTYPE = DESCRIPTOR.enum_types_by_name['InputVertexType']
58    InputVertexType = enum_type_wrapper.EnumTypeWrapper(_INPUTVERTEXTYPE)
59    _OUTPUTVERTEXTYPE = DESCRIPTOR.enum_types_by_name['OutputVertexType']
60    OutputVertexType = enum_type_wrapper.EnumTypeWrapper(_OUTPUTVERTEXTYPE)
61    RUN = 1
62    DATASET = 2
63    MODEL = 3
64    RUN_OUTPUT = 1
65    MODEL_OUTPUT = 2
66  
67  
68    if _descriptor._USE_C_DESCRIPTORS == False:
69  
70      DESCRIPTOR._options = None
71      DESCRIPTOR._serialized_options = b'\n\031org.mlflow.internal.proto\220\001\001\342?\002\020\001'
72      _INPUTVERTEXTYPE._serialized_start=58
73      _INPUTVERTEXTYPE._serialized_end=108
74      _OUTPUTVERTEXTYPE._serialized_start=110
75      _OUTPUTVERTEXTYPE._serialized_end=162
76    # @@protoc_insertion_point(module_scope)
77