| # |
| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| # |
| # -*- coding: utf-8 -*- |
| # Generated by the protocol buffer compiler. DO NOT EDIT! |
| # NO CHECKED-IN PROTOBUF GENCODE |
| # source: spark/connect/ml.proto |
| # Protobuf Python Version: 6.33.0 |
| """Generated protocol buffer code.""" |
| from google.protobuf import descriptor as _descriptor |
| from google.protobuf import descriptor_pool as _descriptor_pool |
| from google.protobuf import runtime_version as _runtime_version |
| from google.protobuf import symbol_database as _symbol_database |
| from google.protobuf.internal import builder as _builder |
| |
| _runtime_version.ValidateProtobufRuntimeVersion( |
| _runtime_version.Domain.PUBLIC, 6, 33, 0, "", "spark/connect/ml.proto" |
| ) |
| # @@protoc_insertion_point(imports) |
| |
| _sym_db = _symbol_database.Default() |
| |
| |
| from pyspark.sql.connect.proto import relations_pb2 as spark_dot_connect_dot_relations__pb2 |
| from pyspark.sql.connect.proto import expressions_pb2 as spark_dot_connect_dot_expressions__pb2 |
| from pyspark.sql.connect.proto import ml_common_pb2 as spark_dot_connect_dot_ml__common__pb2 |
| |
| |
| DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile( |
| b'\n\x16spark/connect/ml.proto\x12\rspark.connect\x1a\x1dspark/connect/relations.proto\x1a\x1fspark/connect/expressions.proto\x1a\x1dspark/connect/ml_common.proto"\xc7\x0e\n\tMlCommand\x12\x30\n\x03\x66it\x18\x01 \x01(\x0b\x32\x1c.spark.connect.MlCommand.FitH\x00R\x03\x66it\x12,\n\x05\x66\x65tch\x18\x02 \x01(\x0b\x32\x14.spark.connect.FetchH\x00R\x05\x66\x65tch\x12\x39\n\x06\x64\x65lete\x18\x03 \x01(\x0b\x32\x1f.spark.connect.MlCommand.DeleteH\x00R\x06\x64\x65lete\x12\x36\n\x05write\x18\x04 \x01(\x0b\x32\x1e.spark.connect.MlCommand.WriteH\x00R\x05write\x12\x33\n\x04read\x18\x05 \x01(\x0b\x32\x1d.spark.connect.MlCommand.ReadH\x00R\x04read\x12?\n\x08\x65valuate\x18\x06 \x01(\x0b\x32!.spark.connect.MlCommand.EvaluateH\x00R\x08\x65valuate\x12\x46\n\x0b\x63lean_cache\x18\x07 \x01(\x0b\x32#.spark.connect.MlCommand.CleanCacheH\x00R\ncleanCache\x12M\n\x0eget_cache_info\x18\x08 \x01(\x0b\x32%.spark.connect.MlCommand.GetCacheInfoH\x00R\x0cgetCacheInfo\x12O\n\x0e\x63reate_summary\x18\t \x01(\x0b\x32&.spark.connect.MlCommand.CreateSummaryH\x00R\rcreateSummary\x12M\n\x0eget_model_size\x18\n \x01(\x0b\x32%.spark.connect.MlCommand.GetModelSizeH\x00R\x0cgetModelSize\x1a\xb2\x01\n\x03\x46it\x12\x37\n\testimator\x18\x01 \x01(\x0b\x32\x19.spark.connect.MlOperatorR\testimator\x12\x34\n\x06params\x18\x02 \x01(\x0b\x32\x17.spark.connect.MlParamsH\x00R\x06params\x88\x01\x01\x12\x31\n\x07\x64\x61taset\x18\x03 \x01(\x0b\x32\x17.spark.connect.RelationR\x07\x64\x61tasetB\t\n\x07_params\x1ap\n\x06\x44\x65lete\x12\x33\n\x08obj_refs\x18\x01 \x03(\x0b\x32\x18.spark.connect.ObjectRefR\x07objRefs\x12"\n\nevict_only\x18\x02 \x01(\x08H\x00R\tevictOnly\x88\x01\x01\x42\r\n\x0b_evict_only\x1a\x0c\n\nCleanCache\x1a\x0e\n\x0cGetCacheInfo\x1a\x9a\x03\n\x05Write\x12\x37\n\x08operator\x18\x01 \x01(\x0b\x32\x19.spark.connect.MlOperatorH\x00R\x08operator\x12\x33\n\x07obj_ref\x18\x02 \x01(\x0b\x32\x18.spark.connect.ObjectRefH\x00R\x06objRef\x12\x34\n\x06params\x18\x03 \x01(\x0b\x32\x17.spark.connect.MlParamsH\x01R\x06params\x88\x01\x01\x12\x12\n\x04path\x18\x04 \x01(\tR\x04path\x12.\n\x10should_overwrite\x18\x05 \x01(\x08H\x02R\x0fshouldOverwrite\x88\x01\x01\x12\x45\n\x07options\x18\x06 \x03(\x0b\x32+.spark.connect.MlCommand.Write.OptionsEntryR\x07options\x1a:\n\x0cOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x06\n\x04typeB\t\n\x07_paramsB\x13\n\x11_should_overwrite\x1aQ\n\x04Read\x12\x35\n\x08operator\x18\x01 \x01(\x0b\x32\x19.spark.connect.MlOperatorR\x08operator\x12\x12\n\x04path\x18\x02 \x01(\tR\x04path\x1a\xb7\x01\n\x08\x45valuate\x12\x37\n\tevaluator\x18\x01 \x01(\x0b\x32\x19.spark.connect.MlOperatorR\tevaluator\x12\x34\n\x06params\x18\x02 \x01(\x0b\x32\x17.spark.connect.MlParamsH\x00R\x06params\x88\x01\x01\x12\x31\n\x07\x64\x61taset\x18\x03 \x01(\x0b\x32\x17.spark.connect.RelationR\x07\x64\x61tasetB\t\n\x07_params\x1ay\n\rCreateSummary\x12\x35\n\tmodel_ref\x18\x01 \x01(\x0b\x32\x18.spark.connect.ObjectRefR\x08modelRef\x12\x31\n\x07\x64\x61taset\x18\x02 \x01(\x0b\x32\x17.spark.connect.RelationR\x07\x64\x61taset\x1a\x45\n\x0cGetModelSize\x12\x35\n\tmodel_ref\x18\x01 \x01(\x0b\x32\x18.spark.connect.ObjectRefR\x08modelRefB\t\n\x07\x63ommand"\xd5\x03\n\x0fMlCommandResult\x12\x39\n\x05param\x18\x01 \x01(\x0b\x32!.spark.connect.Expression.LiteralH\x00R\x05param\x12\x1a\n\x07summary\x18\x02 \x01(\tH\x00R\x07summary\x12T\n\roperator_info\x18\x03 \x01(\x0b\x32-.spark.connect.MlCommandResult.MlOperatorInfoH\x00R\x0coperatorInfo\x1a\x85\x02\n\x0eMlOperatorInfo\x12\x33\n\x07obj_ref\x18\x01 \x01(\x0b\x32\x18.spark.connect.ObjectRefH\x00R\x06objRef\x12\x14\n\x04name\x18\x02 \x01(\tH\x00R\x04name\x12\x15\n\x03uid\x18\x03 \x01(\tH\x01R\x03uid\x88\x01\x01\x12\x34\n\x06params\x18\x04 \x01(\x0b\x32\x17.spark.connect.MlParamsH\x02R\x06params\x88\x01\x01\x12,\n\x0fwarning_message\x18\x05 \x01(\tH\x03R\x0ewarningMessage\x88\x01\x01\x42\x06\n\x04typeB\x06\n\x04_uidB\t\n\x07_paramsB\x12\n\x10_warning_messageB\r\n\x0bresult_typeB6\n\x1eorg.apache.spark.connect.protoP\x01Z\x12internal/generatedb\x06proto3' |
| ) |
| |
| _globals = globals() |
| _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals) |
| _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "pyspark.sql.connect.proto.ml_pb2", _globals) |
| if not _descriptor._USE_C_DESCRIPTORS: |
| _globals["DESCRIPTOR"]._loaded_options = None |
| _globals[ |
| "DESCRIPTOR" |
| ]._serialized_options = b"\n\036org.apache.spark.connect.protoP\001Z\022internal/generated" |
| _globals["_MLCOMMAND_WRITE_OPTIONSENTRY"]._loaded_options = None |
| _globals["_MLCOMMAND_WRITE_OPTIONSENTRY"]._serialized_options = b"8\001" |
| _globals["_MLCOMMAND"]._serialized_start = 137 |
| _globals["_MLCOMMAND"]._serialized_end = 2000 |
| _globals["_MLCOMMAND_FIT"]._serialized_start = 791 |
| _globals["_MLCOMMAND_FIT"]._serialized_end = 969 |
| _globals["_MLCOMMAND_DELETE"]._serialized_start = 971 |
| _globals["_MLCOMMAND_DELETE"]._serialized_end = 1083 |
| _globals["_MLCOMMAND_CLEANCACHE"]._serialized_start = 1085 |
| _globals["_MLCOMMAND_CLEANCACHE"]._serialized_end = 1097 |
| _globals["_MLCOMMAND_GETCACHEINFO"]._serialized_start = 1099 |
| _globals["_MLCOMMAND_GETCACHEINFO"]._serialized_end = 1113 |
| _globals["_MLCOMMAND_WRITE"]._serialized_start = 1116 |
| _globals["_MLCOMMAND_WRITE"]._serialized_end = 1526 |
| _globals["_MLCOMMAND_WRITE_OPTIONSENTRY"]._serialized_start = 1428 |
| _globals["_MLCOMMAND_WRITE_OPTIONSENTRY"]._serialized_end = 1486 |
| _globals["_MLCOMMAND_READ"]._serialized_start = 1528 |
| _globals["_MLCOMMAND_READ"]._serialized_end = 1609 |
| _globals["_MLCOMMAND_EVALUATE"]._serialized_start = 1612 |
| _globals["_MLCOMMAND_EVALUATE"]._serialized_end = 1795 |
| _globals["_MLCOMMAND_CREATESUMMARY"]._serialized_start = 1797 |
| _globals["_MLCOMMAND_CREATESUMMARY"]._serialized_end = 1918 |
| _globals["_MLCOMMAND_GETMODELSIZE"]._serialized_start = 1920 |
| _globals["_MLCOMMAND_GETMODELSIZE"]._serialized_end = 1989 |
| _globals["_MLCOMMANDRESULT"]._serialized_start = 2003 |
| _globals["_MLCOMMANDRESULT"]._serialized_end = 2472 |
| _globals["_MLCOMMANDRESULT_MLOPERATORINFO"]._serialized_start = 2196 |
| _globals["_MLCOMMANDRESULT_MLOPERATORINFO"]._serialized_end = 2457 |
| # @@protoc_insertion_point(module_scope) |
