gimlet-api 0.0.8__py3-none-any.whl → 0.0.10__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {gimlet_api-0.0.8.dist-info → gimlet_api-0.0.10.dist-info}/METADATA +4 -2
- {gimlet_api-0.0.8.dist-info → gimlet_api-0.0.10.dist-info}/RECORD +22 -20
- gml/client.py +29 -20
- gml/compile.py +48 -2
- gml/device.py +2 -0
- gml/hf.py +240 -38
- gml/model.py +37 -0
- gml/pipelines.py +21 -1
- gml/preprocessing.py +17 -3
- gml/proto/opentelemetry/proto/metrics/v1/metrics_pb2.py +39 -38
- gml/proto/src/api/corepb/v1/compiled_pipeline_pb2.py +64 -0
- gml/proto/src/api/corepb/v1/controlplane_pb2.py +60 -29
- gml/proto/src/api/corepb/v1/cp_edge_pb2.py +37 -35
- gml/proto/src/api/corepb/v1/deployed_pipeline_pb2.py +37 -0
- gml/proto/src/api/corepb/v1/device_info_pb2.py +21 -9
- gml/proto/src/api/corepb/v1/gem_config_pb2.py +28 -15
- gml/proto/src/api/corepb/v1/mediastream_pb2.py +42 -39
- gml/proto/src/api/corepb/v1/model_exec_pb2.py +131 -107
- gml/proto/src/controlplane/compiler/cpb/v1/cpb_pb2.py +22 -7
- gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2.py +25 -23
- gml/tensor.py +4 -0
- {gimlet_api-0.0.8.dist-info → gimlet_api-0.0.10.dist-info}/WHEEL +0 -0
@@ -16,7 +16,7 @@ from gml.proto.src.common.typespb import uuid_pb2 as src_dot_common_dot_typespb_
|
|
16
16
|
from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
|
17
17
|
|
18
18
|
|
19
|
-
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3src/controlplane/logicalpipeline/lppb/v1/lppb.proto\x12,gml.internal.controlplane.logicalpipeline.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"
|
19
|
+
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3src/controlplane/logicalpipeline/lppb/v1/lppb.proto\x12,gml.internal.controlplane.logicalpipeline.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\"src/api/corepb/v1/model_exec.proto\"\x9a\x02\n\x13LogicalPipelineInfo\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04yaml\x18\x04 \x01(\tR\x04yaml\x12N\n\rpipeline_spec\x18\x05 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0cpipelineSpec\x12/\n\x0b\x65ncoded_dag\x18\x06 \x01(\tB\x0e\xe2\xde\x1f\nEncodedDAGR\nencodedDag\"P\n\x1bListLogicalPipelinesRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\x7f\n\x1cListLogicalPipelinesResponse\x12_\n\tpipelines\x18\x01 \x03(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\tpipelines\"\x8b\x01\n\x19GetLogicalPipelineRequest\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x31\n\x06org_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\"\x9c\x01\n\x1aGetLogicalPipelineResponse\x12\x66\n\rpipeline_info\x18\x02 \x01(\x0b\x32\x41.gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineInfoR\x0cpipelineInfoJ\x04\x08\x01\x10\x02J\x04\x08\x03\x10\x04J\x04\x08\x04\x10\x05J\x04\x08\x05\x10\x06\"\x7f\n\x1c\x43reateLogicalPipelineRequest\x12\x31\n\x06org_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\t\xe2\xde\x1f\x05OrgIDR\x05orgId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x12\n\x04yaml\x18\x04 \x01(\tR\x04yamlJ\x04\x08\x03\x10\x04\"H\n\x1d\x43reateLogicalPipelineResponse\x12\'\n\x02id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x06\xe2\xde\x1f\x02IDR\x02id\"5\n\x1fParseLogicalPipelineYAMLRequest\x12\x12\n\x04yaml\x18\x01 \x01(\tR\x04yaml\"x\n ParseLogicalPipelineYAMLResponse\x12T\n\x10logical_pipeline\x18\x01 \x01(\x0b\x32).gml.internal.api.core.v1.LogicalPipelineR\x0flogicalPipeline2\xe1\x05\n\x16LogicalPipelineService\x12\xad\x01\n\x14ListLogicalPipelines\x12I.gml.internal.controlplane.logicalpipeline.v1.ListLogicalPipelinesRequest\x1aJ.gml.internal.controlplane.logicalpipeline.v1.ListLogicalPipelinesResponse\x12\xa7\x01\n\x12GetLogicalPipeline\x12G.gml.internal.controlplane.logicalpipeline.v1.GetLogicalPipelineRequest\x1aH.gml.internal.controlplane.logicalpipeline.v1.GetLogicalPipelineResponse\x12\xb0\x01\n\x15\x43reateLogicalPipeline\x12J.gml.internal.controlplane.logicalpipeline.v1.CreateLogicalPipelineRequest\x1aK.gml.internal.controlplane.logicalpipeline.v1.CreateLogicalPipelineResponse\x12\xb9\x01\n\x18ParseLogicalPipelineYAML\x12M.gml.internal.controlplane.logicalpipeline.v1.ParseLogicalPipelineYAMLRequest\x1aN.gml.internal.controlplane.logicalpipeline.v1.ParseLogicalPipelineYAMLResponseBDZBgimletlabs.ai/gimlet/src/controlplane/logicalpipeline/lppb/v1;lppbb\x06proto3')
|
20
20
|
|
21
21
|
_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
|
22
22
|
_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.controlplane.logicalpipeline.lppb.v1.lppb_pb2', globals())
|
@@ -26,36 +26,38 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
26
26
|
DESCRIPTOR._serialized_options = b'ZBgimletlabs.ai/gimlet/src/controlplane/logicalpipeline/lppb/v1;lppb'
|
27
27
|
_LOGICALPIPELINEINFO.fields_by_name['id']._options = None
|
28
28
|
_LOGICALPIPELINEINFO.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
29
|
+
_LOGICALPIPELINEINFO.fields_by_name['org_id']._options = None
|
30
|
+
_LOGICALPIPELINEINFO.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
31
|
+
_LOGICALPIPELINEINFO.fields_by_name['encoded_dag']._options = None
|
32
|
+
_LOGICALPIPELINEINFO.fields_by_name['encoded_dag']._serialized_options = b'\342\336\037\nEncodedDAG'
|
29
33
|
_LISTLOGICALPIPELINESREQUEST.fields_by_name['org_id']._options = None
|
30
34
|
_LISTLOGICALPIPELINESREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
31
35
|
_GETLOGICALPIPELINEREQUEST.fields_by_name['id']._options = None
|
32
36
|
_GETLOGICALPIPELINEREQUEST.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
33
37
|
_GETLOGICALPIPELINEREQUEST.fields_by_name['org_id']._options = None
|
34
38
|
_GETLOGICALPIPELINEREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
35
|
-
_GETLOGICALPIPELINERESPONSE.fields_by_name['encoded_dag']._options = None
|
36
|
-
_GETLOGICALPIPELINERESPONSE.fields_by_name['encoded_dag']._serialized_options = b'\342\336\037\nEncodedDAG'
|
37
39
|
_CREATELOGICALPIPELINEREQUEST.fields_by_name['org_id']._options = None
|
38
40
|
_CREATELOGICALPIPELINEREQUEST.fields_by_name['org_id']._serialized_options = b'\342\336\037\005OrgID'
|
39
41
|
_CREATELOGICALPIPELINERESPONSE.fields_by_name['id']._options = None
|
40
42
|
_CREATELOGICALPIPELINERESPONSE.fields_by_name['id']._serialized_options = b'\342\336\037\002ID'
|
41
|
-
_LOGICALPIPELINEINFO._serialized_start=
|
42
|
-
_LOGICALPIPELINEINFO._serialized_end=
|
43
|
-
_LISTLOGICALPIPELINESREQUEST._serialized_start=
|
44
|
-
_LISTLOGICALPIPELINESREQUEST._serialized_end=
|
45
|
-
_LISTLOGICALPIPELINESRESPONSE._serialized_start=
|
46
|
-
_LISTLOGICALPIPELINESRESPONSE._serialized_end=
|
47
|
-
_GETLOGICALPIPELINEREQUEST._serialized_start=
|
48
|
-
_GETLOGICALPIPELINEREQUEST._serialized_end=
|
49
|
-
_GETLOGICALPIPELINERESPONSE._serialized_start=
|
50
|
-
_GETLOGICALPIPELINERESPONSE._serialized_end=
|
51
|
-
_CREATELOGICALPIPELINEREQUEST._serialized_start=
|
52
|
-
_CREATELOGICALPIPELINEREQUEST._serialized_end=
|
53
|
-
_CREATELOGICALPIPELINERESPONSE._serialized_start=
|
54
|
-
_CREATELOGICALPIPELINERESPONSE._serialized_end=
|
55
|
-
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_start=
|
56
|
-
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_end=
|
57
|
-
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_start=
|
58
|
-
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_end=
|
59
|
-
_LOGICALPIPELINESERVICE._serialized_start=
|
60
|
-
_LOGICALPIPELINESERVICE._serialized_end=
|
43
|
+
_LOGICALPIPELINEINFO._serialized_start=191
|
44
|
+
_LOGICALPIPELINEINFO._serialized_end=473
|
45
|
+
_LISTLOGICALPIPELINESREQUEST._serialized_start=475
|
46
|
+
_LISTLOGICALPIPELINESREQUEST._serialized_end=555
|
47
|
+
_LISTLOGICALPIPELINESRESPONSE._serialized_start=557
|
48
|
+
_LISTLOGICALPIPELINESRESPONSE._serialized_end=684
|
49
|
+
_GETLOGICALPIPELINEREQUEST._serialized_start=687
|
50
|
+
_GETLOGICALPIPELINEREQUEST._serialized_end=826
|
51
|
+
_GETLOGICALPIPELINERESPONSE._serialized_start=829
|
52
|
+
_GETLOGICALPIPELINERESPONSE._serialized_end=985
|
53
|
+
_CREATELOGICALPIPELINEREQUEST._serialized_start=987
|
54
|
+
_CREATELOGICALPIPELINEREQUEST._serialized_end=1114
|
55
|
+
_CREATELOGICALPIPELINERESPONSE._serialized_start=1116
|
56
|
+
_CREATELOGICALPIPELINERESPONSE._serialized_end=1188
|
57
|
+
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_start=1190
|
58
|
+
_PARSELOGICALPIPELINEYAMLREQUEST._serialized_end=1243
|
59
|
+
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_start=1245
|
60
|
+
_PARSELOGICALPIPELINEYAMLRESPONSE._serialized_end=1365
|
61
|
+
_LOGICALPIPELINESERVICE._serialized_start=1368
|
62
|
+
_LOGICALPIPELINESERVICE._serialized_end=2105
|
61
63
|
# @@protoc_insertion_point(module_scope)
|
gml/tensor.py
CHANGED
@@ -77,6 +77,10 @@ def chan_format_str_to_proto(chan_format: str):
|
|
77
77
|
return (
|
78
78
|
modelexecpb.DimensionSemantics.ImageChannelParams.IMAGE_CHANNEL_FORMAT_BGR
|
79
79
|
)
|
80
|
+
case "gray":
|
81
|
+
return (
|
82
|
+
modelexecpb.DimensionSemantics.ImageChannelParams.IMAGE_CHANNEL_FORMAT_GRAYSCALE
|
83
|
+
)
|
80
84
|
case _:
|
81
85
|
raise ValueError("Invalid channel_format format: {}".format(chan_format))
|
82
86
|
|
File without changes
|