gimlet-api 0.0.13__py3-none-any.whl → 0.0.15__py3-none-any.whl

gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py

@@ -29,6 +29,11 @@ class LogicalPipelineServiceStub(object):
                 request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineRequest.SerializeToString,
                 response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineResponse.FromString,
                 )
+        self.UpdateLogicalPipeline = channel.unary_unary(
+                '/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/UpdateLogicalPipeline',
+                request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.UpdateLogicalPipelineRequest.SerializeToString,
+                response_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.UpdateLogicalPipelineResponse.FromString,
+                )
         self.ParseLogicalPipelineYAML = channel.unary_unary(
                 '/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/ParseLogicalPipelineYAML',
                 request_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.SerializeToString,
@@ -62,6 +67,12 @@ class LogicalPipelineServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def UpdateLogicalPipeline(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def ParseLogicalPipelineYAML(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -92,6 +103,11 @@ def add_LogicalPipelineServiceServicer_to_server(servicer, server):
                     request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineRequest.FromString,
                     response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.CreateLogicalPipelineResponse.SerializeToString,
             ),
+            'UpdateLogicalPipeline': grpc.unary_unary_rpc_method_handler(
+                    servicer.UpdateLogicalPipeline,
+                    request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.UpdateLogicalPipelineRequest.FromString,
+                    response_serializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.UpdateLogicalPipelineResponse.SerializeToString,
+            ),
             'ParseLogicalPipelineYAML': grpc.unary_unary_rpc_method_handler(
                     servicer.ParseLogicalPipelineYAML,
                     request_deserializer=src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.ParseLogicalPipelineYAMLRequest.FromString,
@@ -163,6 +179,23 @@ class LogicalPipelineService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def UpdateLogicalPipeline(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/gml.internal.controlplane.logicalpipeline.v1.LogicalPipelineService/UpdateLogicalPipeline',
+            src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.UpdateLogicalPipelineRequest.SerializeToString,
+            src_dot_controlplane_dot_logicalpipeline_dot_lppb_dot_v1_dot_lppb__pb2.UpdateLogicalPipelineResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def ParseLogicalPipelineYAML(request,
             target,

gml/tensor.py

@@ -278,6 +278,13 @@ class EmbeddingDimension(DimensionSemantics):
         )
 
 
+class PositionIDsDimension(DimensionSemantics):
+    def to_proto(self) -> modelexecpb.DimensionSemantics:
+        return modelexecpb.DimensionSemantics(
+            kind=modelexecpb.DimensionSemantics.DIMENSION_SEMANTICS_KIND_POSITION_IDS,
+        )
+
+
 class TensorSemantics:
     def __init__(
         self,