frogml-core 0.0.103__py3-none-any.whl → 0.0.104__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.
frogml_core/__init__.py CHANGED
@@ -1,7 +1,7 @@
1
1
  """Top-level package for frogml."""
2
2
 
3
3
  __author__ = "jfrog"
4
- __version__ = "0.0.103"
4
+ __version__ = "0.0.104"
5
5
 
6
6
  from frogml_core.inner.di_configuration import wire_dependencies
7
7
  from frogml_core.model.model_version_tracking import ( # noqa: F401,E501
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: frogml-core
3
- Version: 0.0.103
3
+ Version: 0.0.104
4
4
  Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
5
5
  License: Apache-2.0
6
6
  Keywords: mlops,ml,deployment,serving,model
@@ -1,4 +1,4 @@
1
- frogml_core/__init__.py,sha256=br4ZxX0FM7gtpFHPynfXrMRBS5nr4-xOQecddpJZ8GM,778
1
+ frogml_core/__init__.py,sha256=BiKZixjvvj4ovlXfOsqqz7fjjZDn5gmtm8qmPXGjNTw,778
2
2
  frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
3
3
  frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
4
4
  frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -917,8 +917,8 @@ frogml_proto/qwak/kube_deployment_captain/alert_pb2_grpc.py,sha256=1oboBPFxaTEXt
917
917
  frogml_proto/qwak/kube_deployment_captain/alerting_pb2.py,sha256=2WquRLoQXoQMVRn4q0eK4ouA0bbOFUeXBYqqhuUER4s,3717
918
918
  frogml_proto/qwak/kube_deployment_captain/alerting_pb2.pyi,sha256=AGXtot0uXW-mGd2xuixCCJB6eTt0UuF3b2oAhb4IsNc,3759
919
919
  frogml_proto/qwak/kube_deployment_captain/alerting_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
920
- frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.py,sha256=v6jo-fsMN7uLW_i5MCiMWDQZ8bsmtB4oCdwi4DtHuZo,29289
921
- frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi,sha256=e5qiXwffNn4kDUz1dM61o8f8r0v-cB9PwBedM37rYRM,39524
920
+ frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.py,sha256=btbJlJfan4bao4ktbvviClA6H8To5H9MFsECk8G7B-4,29136
921
+ frogml_proto/qwak/kube_deployment_captain/batch_job_pb2.pyi,sha256=FBm9WayjC-RJgF76Kw7zgaajKsoVinXAqpN4LfW6M08,39300
922
922
  frogml_proto/qwak/kube_deployment_captain/batch_job_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
923
923
  frogml_proto/qwak/kube_deployment_captain/deployment_pb2.py,sha256=3u38KAg__KYW1n-YlLDQCJxTDXj7eyQSbILEoX7fsz8,14050
924
924
  frogml_proto/qwak/kube_deployment_captain/deployment_pb2.pyi,sha256=Sz11iTrL3BlUEl1jXNf0Q1SDe1SX_MjjyB9bMrtsOVg,24090
@@ -1114,7 +1114,7 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
1114
1114
  frogml_services_mock/services_mock.py,sha256=sgKgwhu2W0YOHtzil8x7f1znK_sZr_i27XSeiF4xqVE,21200
1115
1115
  frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1116
1116
  frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
1117
- frogml_storage/__init__.py,sha256=6Qw9hRq6csqdj5snF0TysZZtzmHEBPYB_RC0hwEZRW0,24
1117
+ frogml_storage/__init__.py,sha256=m9tOf0gVCkqN3dmK4Ruh3l4dO8KbQ-WtCnvEBME8wSU,24
1118
1118
  frogml_storage/_environment.py,sha256=zuzOJBtBwFaguwn_JkKjfhXStZoustgP30KzOP3mYv8,707
1119
1119
  frogml_storage/artifactory/__init__.py,sha256=C02rcm7kqsZBVA6c6Gztxamj96hn8Aj6BuzYWFRmWbQ,71
1120
1120
  frogml_storage/artifactory/_artifactory_api.py,sha256=Oz0HOpQPSNwWIVAy94UJUyPhLetc7sdZjoTfSXtrFug,11200
@@ -1153,6 +1153,6 @@ frogml_storage/utils/__init__.py,sha256=HQUWfuGUIPZY7kfS795TRW8BQ4WmNqrNjS7lUrbx
1153
1153
  frogml_storage/utils/_input_checks_utility.py,sha256=CFiJOdTBS9piJMtR3lemEz27wZcQ6_-7XESu8iy-mrw,3221
1154
1154
  frogml_storage/utils/_storage_utils.py,sha256=HB2g7uY5A3b33yIcAUM1OjHb5jWsnpESsiDrEviQwrI,366
1155
1155
  frogml_storage/utils/_url_utils.py,sha256=NUEfz9Fp1iE8b676-A5wrMlSTsJVRKrUhcUItOFAJD8,821
1156
- frogml_core-0.0.103.dist-info/METADATA,sha256=Oau3MMkXSXA3l9pfuHs3kaNyT722h6WVjsNeErS_FNA,14976
1157
- frogml_core-0.0.103.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
1158
- frogml_core-0.0.103.dist-info/RECORD,,
1156
+ frogml_core-0.0.104.dist-info/METADATA,sha256=I5Dv5QvSm8_PJ5QP1ASF4UaT0A8ZJi6Q0og06u7TV_w,14976
1157
+ frogml_core-0.0.104.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
1158
+ frogml_core-0.0.104.dist-info/RECORD,,
@@ -13,10 +13,9 @@ from google.protobuf import symbol_database as _symbol_database
13
13
  _sym_db = _symbol_database.Default()
14
14
 
15
15
 
16
- from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
17
16
 
18
17
 
19
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/kube_deployment_captain/batch_job.proto\x12\x1cqwak.kube.deployment.captain\x1a\x1egoogle/protobuf/wrappers.proto\"\xd2\x03\n\x1cListInferenceJobFilesRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x16\n\x0e\x64irectory_path\x18\x02 \x01(\t\x12\x18\n\x0ctoken_secret\x18\x03 \x01(\tB\x02\x18\x01\x12\x19\n\rsecret_secret\x18\x04 \x01(\tB\x02\x18\x01\x12\x1a\n\x12secret_service_url\x18\x05 \x01(\t\x12\x14\n\x08role_arn\x18\x06 \x01(\tB\x02\x18\x01\x12G\n\x0f\x61ws_credentials\x18\x07 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x08 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12P\n\x14grouped_task_details\x18\t \x01(\x0b\x32\x30.qwak.kube.deployment.captain.GroupedTaskDetailsH\x01\x12\x16\n\x0e\x65nvironment_id\x18\n \x01(\tB\x1a\n\x18\x63loud_client_credentialsB\x0b\n\ttask_flow\"c\n\x12GroupedTaskDetails\x12\x17\n\x0f\x63oncurrentTasks\x18\x01 \x01(\x05\x12\x18\n\x10\x64\x65stination_path\x18\x02 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x03 \x01(\t\"\xb4\x02\n)CleanStorageObjectFromCloudStorageRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x15\n\robject_prefix\x18\x02 \x03(\t\x12\x1a\n\x12secret_service_url\x18\x03 \x01(\t\x12G\n\x0f\x61ws_credentials\x18\x04 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x05 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12\x16\n\x0e\x65nvironment_id\x18\x06 \x01(\tB\x1a\n\x18\x63loud_client_credentials\"O\n\x0e\x41wsCredentials\x12\x10\n\x08role_arn\x18\x01 \x01(\t\x12\x14\n\x0ctoken_secret\x18\x02 \x01(\t\x12\x15\n\rsecret_secret\x18\x04 \x01(\t\"9\n\x0eGcpCredentials\x12\'\n\x1fservice_account_json_key_secret\x18\x01 \x01(\t\"\xa6\x01\n\x1dListInferenceJobFilesResponse\x12\x12\n\nfile_names\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x03 \x01(\t\x12H\n\x11list_file_details\x18\x04 \x03(\x0b\x32-.qwak.kube.deployment.captain.ListFileDetails\"W\n\x0fListFileDetails\x12\x15\n\ris_point_file\x18\x01 \x01(\x08\x12\x17\n\x0fpoint_file_path\x18\x02 \x01(\t\x12\x14\n\x0cobject_paths\x18\x03 \x03(\t\"\xf0\x01\n\"CreateInferenceTaskExecutorRequest\x12\x63\n\x1btask_executor_configuration\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.TaskExecutorConfigurationMessage\x12\x65\n\x1cinference_task_configuration\x18\x02 \x01(\x0b\x32?.qwak.kube.deployment.captain.InferenceTaskConfigurationMessage\"\x89\x01\n\x1aPrepareInferenceJobRequest\x12k\n#inference_job_configuration_message\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.InferenceJobConfigurationMessage\"F\n\x1bPrepareInferenceJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xcd\x04\n TaskExecutorConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\x12G\n\x10model_identifier\x18\x05 \x01(\x0b\x32-.qwak.kube.deployment.captain.ModelIdentifier\x12\x11\n\timage_url\x18\x06 \x01(\t\x12\x15\n\rbackoff_limit\x18\x07 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x08 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\t \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x0c \x01(\t\x12\x41\n\x08job_size\x18\r \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x17\n\x0fpurchase_option\x18\x0e \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0f \x01(\t\x12%\n\x1d\x63ustom_service_account_secret\x18\x10 \x01(\t\x12#\n\x1bjfrog_token_api_secret_name\x18\x11 \x01(\t\x12\x16\n\x0ejfrog_base_url\x18\x12 \x01(\t\"q\n InferenceJobConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x03 \x01(\t\"`\n\x0fModelIdentifier\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\xa2\x01\n\x1dListFilesPreStepConfiguration\x12\x13\n\x0bsource_path\x18\x01 \x01(\t\x12\x1c\n\x14list_files_file_name\x18\x02 \x01(\t\x12\x1d\n\x15total_number_of_tasks\x18\x03 \x01(\x05\x12/\n\ntask_index\x18\x04 \x01(\x0b\x32\x1b.google.protobuf.Int32Value\"\xf4\x04\n!InferenceTaskConfigurationMessage\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x10\n\x08\x66ilepath\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x44\n\x0finput_file_type\x18\x05 \x01(\x0e\x32+.qwak.kube.deployment.captain.InputFileType\x12\x46\n\x10output_file_type\x18\x06 \x01(\x0e\x32,.qwak.kube.deployment.captain.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x07 \x01(\t\x12\x15\n\rsecret_secret\x18\x08 \x01(\t\x12\x43\n\nparameters\x18\t \x03(\x0b\x32/.qwak.kube.deployment.captain.BatchJobParameter\x12\x43\n\x0e\x63loud_provider\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.CloudProvider\x12\'\n\x1fservice_account_json_key_secret\x18\x0b \x01(\t\x12\x1a\n\x12is_point_file_path\x18\x0c \x01(\x08\x12\x66\n!list_files_pre_step_configuration\x18\r \x01(\x0b\x32;.qwak.kube.deployment.captain.ListFilesPreStepConfiguration\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"?\n#CleanInferenceTasksExecutorsRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\"O\n$CleanInferenceTasksExecutorsResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"X\n!CleanInferenceTaskExecutorRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\"M\n\"CleanInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"N\n#CreateInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xf7\x02\n\x1eStartInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x11\n\timage_url\x18\x04 \x01(\t\x12\x0f\n\x03\x63pu\x18\x05 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x06 \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\x07 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x11\n\texecutors\x18\x08 \x01(\x05\x12\x0f\n\x07timeout\x18\t \x01(\x05\x12\x41\n\x08job_size\x18\n \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x12\n\nmodel_uuid\x18\x0b \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0c \x01(\t\"!\n\x1fStartInferenceJobWarmupResponse\"p\n\x1f\x43\x61ncelInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\"\n CancelInferenceJobWarmupResponse\"\xd5\x01\n\x11\x42\x61tchJobResources\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0b\n\x03\x63pu\x18\x02 \x01(\x02\x12\x15\n\rmemory_amount\x18\x03 \x01(\x05\x12\x41\n\x0cmemory_units\x18\x04 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApi\x12\x41\n\rgpu_resources\x18\x05 \x01(\x0b\x32*.qwak.kube.deployment.captain.GpuResources\"[\n\x0cGpuResources\x12\x37\n\x08gpu_type\x18\x01 \x01(\x0e\x32%.qwak.kube.deployment.captain.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05*:\n\rMemoryUnitApi\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x81\x01\n\rInputFileType\x12\x1d\n\x19UNDEFINED_INPUT_FILE_TYPE\x10\x00\x12\x17\n\x13\x43SV_INPUT_FILE_TYPE\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_INPUT_FILE_TYPE\x10\x02\x12\x1b\n\x17PARQUET_INPUT_FILE_TYPE\x10\x03*\x86\x01\n\x0eOutputFileType\x12\x1e\n\x1aUNDEFINED_OUTPUT_FILE_TYPE\x10\x00\x12\x18\n\x14\x43SV_OUTPUT_FILE_TYPE\x10\x01\x12\x1c\n\x18\x46\x45\x41THER_OUTPUT_FILE_TYPE\x10\x02\x12\x1c\n\x18PARQUET_OUTPUT_FILE_TYPE\x10\x03*\x90\x02\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c*=\n\rCloudProvider\x12\x1a\n\x16UNKNOWN_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
18
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/kube_deployment_captain/batch_job.proto\x12\x1cqwak.kube.deployment.captain\"\xd2\x03\n\x1cListInferenceJobFilesRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x16\n\x0e\x64irectory_path\x18\x02 \x01(\t\x12\x18\n\x0ctoken_secret\x18\x03 \x01(\tB\x02\x18\x01\x12\x19\n\rsecret_secret\x18\x04 \x01(\tB\x02\x18\x01\x12\x1a\n\x12secret_service_url\x18\x05 \x01(\t\x12\x14\n\x08role_arn\x18\x06 \x01(\tB\x02\x18\x01\x12G\n\x0f\x61ws_credentials\x18\x07 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x08 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12P\n\x14grouped_task_details\x18\t \x01(\x0b\x32\x30.qwak.kube.deployment.captain.GroupedTaskDetailsH\x01\x12\x16\n\x0e\x65nvironment_id\x18\n \x01(\tB\x1a\n\x18\x63loud_client_credentialsB\x0b\n\ttask_flow\"c\n\x12GroupedTaskDetails\x12\x17\n\x0f\x63oncurrentTasks\x18\x01 \x01(\x05\x12\x18\n\x10\x64\x65stination_path\x18\x02 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x03 \x01(\t\"\xb4\x02\n)CleanStorageObjectFromCloudStorageRequest\x12\x0e\n\x06\x62ucket\x18\x01 \x01(\t\x12\x15\n\robject_prefix\x18\x02 \x03(\t\x12\x1a\n\x12secret_service_url\x18\x03 \x01(\t\x12G\n\x0f\x61ws_credentials\x18\x04 \x01(\x0b\x32,.qwak.kube.deployment.captain.AwsCredentialsH\x00\x12G\n\x0fgcp_credentials\x18\x05 \x01(\x0b\x32,.qwak.kube.deployment.captain.GcpCredentialsH\x00\x12\x16\n\x0e\x65nvironment_id\x18\x06 \x01(\tB\x1a\n\x18\x63loud_client_credentials\"O\n\x0e\x41wsCredentials\x12\x10\n\x08role_arn\x18\x01 \x01(\t\x12\x14\n\x0ctoken_secret\x18\x02 \x01(\t\x12\x15\n\rsecret_secret\x18\x04 \x01(\t\"9\n\x0eGcpCredentials\x12\'\n\x1fservice_account_json_key_secret\x18\x01 \x01(\t\"\xa6\x01\n\x1dListInferenceJobFilesResponse\x12\x12\n\nfile_names\x18\x01 \x03(\t\x12\x0f\n\x07success\x18\x02 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x03 \x01(\t\x12H\n\x11list_file_details\x18\x04 \x03(\x0b\x32-.qwak.kube.deployment.captain.ListFileDetails\"W\n\x0fListFileDetails\x12\x15\n\ris_point_file\x18\x01 \x01(\x08\x12\x17\n\x0fpoint_file_path\x18\x02 \x01(\t\x12\x14\n\x0cobject_paths\x18\x03 \x03(\t\"\xf0\x01\n\"CreateInferenceTaskExecutorRequest\x12\x63\n\x1btask_executor_configuration\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.TaskExecutorConfigurationMessage\x12\x65\n\x1cinference_task_configuration\x18\x02 \x01(\x0b\x32?.qwak.kube.deployment.captain.InferenceTaskConfigurationMessage\"\x89\x01\n\x1aPrepareInferenceJobRequest\x12k\n#inference_job_configuration_message\x18\x01 \x01(\x0b\x32>.qwak.kube.deployment.captain.InferenceJobConfigurationMessage\"F\n\x1bPrepareInferenceJobResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xcd\x04\n TaskExecutorConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\x12G\n\x10model_identifier\x18\x05 \x01(\x0b\x32-.qwak.kube.deployment.captain.ModelIdentifier\x12\x11\n\timage_url\x18\x06 \x01(\t\x12\x15\n\rbackoff_limit\x18\x07 \x01(\x05\x12\x0f\n\x03\x63pu\x18\x08 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\t \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x16\n\x0e\x65nvironment_id\x18\x0b \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x0c \x01(\t\x12\x41\n\x08job_size\x18\r \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x17\n\x0fpurchase_option\x18\x0e \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0f \x01(\t\x12%\n\x1d\x63ustom_service_account_secret\x18\x10 \x01(\t\x12#\n\x1bjfrog_token_api_secret_name\x18\x11 \x01(\t\x12\x16\n\x0ejfrog_base_url\x18\x12 \x01(\t\"q\n InferenceJobConfigurationMessage\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\x12\x1b\n\x13\x63ustom_iam_role_arn\x18\x03 \x01(\t\"`\n\x0fModelIdentifier\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12\x15\n\tbranch_id\x18\x03 \x01(\tB\x02\x18\x01\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\x85\x01\n\x1dListFilesPreStepConfiguration\x12\x13\n\x0bsource_path\x18\x01 \x01(\t\x12\x1c\n\x14list_files_file_name\x18\x02 \x01(\t\x12\x1d\n\x15total_number_of_tasks\x18\x03 \x01(\x05\x12\x12\n\ntask_index\x18\x04 \x01(\x05\"\xf4\x04\n!InferenceTaskConfigurationMessage\x12\x15\n\rsource_bucket\x18\x01 \x01(\t\x12\x1a\n\x12\x64\x65stination_bucket\x18\x02 \x01(\t\x12\x10\n\x08\x66ilepath\x18\x03 \x01(\t\x12\x18\n\x10\x64\x65stination_path\x18\x04 \x01(\t\x12\x44\n\x0finput_file_type\x18\x05 \x01(\x0e\x32+.qwak.kube.deployment.captain.InputFileType\x12\x46\n\x10output_file_type\x18\x06 \x01(\x0e\x32,.qwak.kube.deployment.captain.OutputFileType\x12\x14\n\x0ctoken_secret\x18\x07 \x01(\t\x12\x15\n\rsecret_secret\x18\x08 \x01(\t\x12\x43\n\nparameters\x18\t \x03(\x0b\x32/.qwak.kube.deployment.captain.BatchJobParameter\x12\x43\n\x0e\x63loud_provider\x18\n \x01(\x0e\x32+.qwak.kube.deployment.captain.CloudProvider\x12\'\n\x1fservice_account_json_key_secret\x18\x0b \x01(\t\x12\x1a\n\x12is_point_file_path\x18\x0c \x01(\x08\x12\x66\n!list_files_pre_step_configuration\x18\r \x01(\x0b\x32;.qwak.kube.deployment.captain.ListFilesPreStepConfiguration\"/\n\x11\x42\x61tchJobParameter\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"?\n#CleanInferenceTasksExecutorsRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\"O\n$CleanInferenceTasksExecutorsResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"X\n!CleanInferenceTaskExecutorRequest\x12\x18\n\x10inference_job_id\x18\x01 \x01(\t\x12\x19\n\x11inference_task_id\x18\x02 \x01(\t\"M\n\"CleanInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"N\n#CreateInferenceTaskExecutorResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x16\n\x0e\x66\x61ilure_reason\x18\x02 \x01(\t\"\xf7\x02\n\x1eStartInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x11\n\timage_url\x18\x04 \x01(\t\x12\x0f\n\x03\x63pu\x18\x05 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x06 \x01(\x05\x42\x02\x18\x01\x12\x45\n\x0cmemory_units\x18\x07 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApiB\x02\x18\x01\x12\x11\n\texecutors\x18\x08 \x01(\x05\x12\x0f\n\x07timeout\x18\t \x01(\x05\x12\x41\n\x08job_size\x18\n \x01(\x0b\x32/.qwak.kube.deployment.captain.BatchJobResources\x12\x12\n\nmodel_uuid\x18\x0b \x01(\t\x12\x19\n\x11image_pull_secret\x18\x0c \x01(\t\"!\n\x1fStartInferenceJobWarmupResponse\"p\n\x1f\x43\x61ncelInferenceJobWarmupRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\"\"\n CancelInferenceJobWarmupResponse\"\xd5\x01\n\x11\x42\x61tchJobResources\x12\x16\n\x0enumber_of_pods\x18\x01 \x01(\x05\x12\x0b\n\x03\x63pu\x18\x02 \x01(\x02\x12\x15\n\rmemory_amount\x18\x03 \x01(\x05\x12\x41\n\x0cmemory_units\x18\x04 \x01(\x0e\x32+.qwak.kube.deployment.captain.MemoryUnitApi\x12\x41\n\rgpu_resources\x18\x05 \x01(\x0b\x32*.qwak.kube.deployment.captain.GpuResources\"[\n\x0cGpuResources\x12\x37\n\x08gpu_type\x18\x01 \x01(\x0e\x32%.qwak.kube.deployment.captain.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05*:\n\rMemoryUnitApi\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x81\x01\n\rInputFileType\x12\x1d\n\x19UNDEFINED_INPUT_FILE_TYPE\x10\x00\x12\x17\n\x13\x43SV_INPUT_FILE_TYPE\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_INPUT_FILE_TYPE\x10\x02\x12\x1b\n\x17PARQUET_INPUT_FILE_TYPE\x10\x03*\x86\x01\n\x0eOutputFileType\x12\x1e\n\x1aUNDEFINED_OUTPUT_FILE_TYPE\x10\x00\x12\x18\n\x14\x43SV_OUTPUT_FILE_TYPE\x10\x01\x12\x1c\n\x18\x46\x45\x41THER_OUTPUT_FILE_TYPE\x10\x02\x12\x1c\n\x18PARQUET_OUTPUT_FILE_TYPE\x10\x03*\x90\x02\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c*=\n\rCloudProvider\x12\x1a\n\x16UNKNOWN_CLOUD_PROVIDER\x10\x00\x12\x07\n\x03\x41WS\x10\x01\x12\x07\n\x03GCP\x10\x02\x42+\n\'com.qwak.ai.kube.deployment.captain.apiP\x01\x62\x06proto3')
20
19
 
21
20
  _MEMORYUNITAPI = DESCRIPTOR.enum_types_by_name['MemoryUnitApi']
22
21
  MemoryUnitApi = enum_type_wrapper.EnumTypeWrapper(_MEMORYUNITAPI)
@@ -301,68 +300,68 @@ if _descriptor._USE_C_DESCRIPTORS == False:
301
300
  _STARTINFERENCEJOBWARMUPREQUEST.fields_by_name['memory_units']._serialized_options = b'\030\001'
302
301
  _CANCELINFERENCEJOBWARMUPREQUEST.fields_by_name['branch_id']._options = None
303
302
  _CANCELINFERENCEJOBWARMUPREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
304
- _MEMORYUNITAPI._serialized_start=4761
305
- _MEMORYUNITAPI._serialized_end=4819
306
- _INPUTFILETYPE._serialized_start=4822
307
- _INPUTFILETYPE._serialized_end=4951
308
- _OUTPUTFILETYPE._serialized_start=4954
309
- _OUTPUTFILETYPE._serialized_end=5088
310
- _GPUTYPE._serialized_start=5091
311
- _GPUTYPE._serialized_end=5363
312
- _CLOUDPROVIDER._serialized_start=5365
313
- _CLOUDPROVIDER._serialized_end=5426
314
- _LISTINFERENCEJOBFILESREQUEST._serialized_start=111
315
- _LISTINFERENCEJOBFILESREQUEST._serialized_end=577
316
- _GROUPEDTASKDETAILS._serialized_start=579
317
- _GROUPEDTASKDETAILS._serialized_end=678
318
- _CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST._serialized_start=681
319
- _CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST._serialized_end=989
320
- _AWSCREDENTIALS._serialized_start=991
321
- _AWSCREDENTIALS._serialized_end=1070
322
- _GCPCREDENTIALS._serialized_start=1072
323
- _GCPCREDENTIALS._serialized_end=1129
324
- _LISTINFERENCEJOBFILESRESPONSE._serialized_start=1132
325
- _LISTINFERENCEJOBFILESRESPONSE._serialized_end=1298
326
- _LISTFILEDETAILS._serialized_start=1300
327
- _LISTFILEDETAILS._serialized_end=1387
328
- _CREATEINFERENCETASKEXECUTORREQUEST._serialized_start=1390
329
- _CREATEINFERENCETASKEXECUTORREQUEST._serialized_end=1630
330
- _PREPAREINFERENCEJOBREQUEST._serialized_start=1633
331
- _PREPAREINFERENCEJOBREQUEST._serialized_end=1770
332
- _PREPAREINFERENCEJOBRESPONSE._serialized_start=1772
333
- _PREPAREINFERENCEJOBRESPONSE._serialized_end=1842
334
- _TASKEXECUTORCONFIGURATIONMESSAGE._serialized_start=1845
335
- _TASKEXECUTORCONFIGURATIONMESSAGE._serialized_end=2434
336
- _INFERENCEJOBCONFIGURATIONMESSAGE._serialized_start=2436
337
- _INFERENCEJOBCONFIGURATIONMESSAGE._serialized_end=2549
338
- _MODELIDENTIFIER._serialized_start=2551
339
- _MODELIDENTIFIER._serialized_end=2647
340
- _LISTFILESPRESTEPCONFIGURATION._serialized_start=2650
341
- _LISTFILESPRESTEPCONFIGURATION._serialized_end=2812
342
- _INFERENCETASKCONFIGURATIONMESSAGE._serialized_start=2815
343
- _INFERENCETASKCONFIGURATIONMESSAGE._serialized_end=3443
344
- _BATCHJOBPARAMETER._serialized_start=3445
345
- _BATCHJOBPARAMETER._serialized_end=3492
346
- _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_start=3494
347
- _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_end=3557
348
- _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_start=3559
349
- _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_end=3638
350
- _CLEANINFERENCETASKEXECUTORREQUEST._serialized_start=3640
351
- _CLEANINFERENCETASKEXECUTORREQUEST._serialized_end=3728
352
- _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_start=3730
353
- _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_end=3807
354
- _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_start=3809
355
- _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_end=3887
356
- _STARTINFERENCEJOBWARMUPREQUEST._serialized_start=3890
357
- _STARTINFERENCEJOBWARMUPREQUEST._serialized_end=4265
358
- _STARTINFERENCEJOBWARMUPRESPONSE._serialized_start=4267
359
- _STARTINFERENCEJOBWARMUPRESPONSE._serialized_end=4300
360
- _CANCELINFERENCEJOBWARMUPREQUEST._serialized_start=4302
361
- _CANCELINFERENCEJOBWARMUPREQUEST._serialized_end=4414
362
- _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_start=4416
363
- _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_end=4450
364
- _BATCHJOBRESOURCES._serialized_start=4453
365
- _BATCHJOBRESOURCES._serialized_end=4666
366
- _GPURESOURCES._serialized_start=4668
367
- _GPURESOURCES._serialized_end=4759
303
+ _MEMORYUNITAPI._serialized_start=4700
304
+ _MEMORYUNITAPI._serialized_end=4758
305
+ _INPUTFILETYPE._serialized_start=4761
306
+ _INPUTFILETYPE._serialized_end=4890
307
+ _OUTPUTFILETYPE._serialized_start=4893
308
+ _OUTPUTFILETYPE._serialized_end=5027
309
+ _GPUTYPE._serialized_start=5030
310
+ _GPUTYPE._serialized_end=5302
311
+ _CLOUDPROVIDER._serialized_start=5304
312
+ _CLOUDPROVIDER._serialized_end=5365
313
+ _LISTINFERENCEJOBFILESREQUEST._serialized_start=79
314
+ _LISTINFERENCEJOBFILESREQUEST._serialized_end=545
315
+ _GROUPEDTASKDETAILS._serialized_start=547
316
+ _GROUPEDTASKDETAILS._serialized_end=646
317
+ _CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST._serialized_start=649
318
+ _CLEANSTORAGEOBJECTFROMCLOUDSTORAGEREQUEST._serialized_end=957
319
+ _AWSCREDENTIALS._serialized_start=959
320
+ _AWSCREDENTIALS._serialized_end=1038
321
+ _GCPCREDENTIALS._serialized_start=1040
322
+ _GCPCREDENTIALS._serialized_end=1097
323
+ _LISTINFERENCEJOBFILESRESPONSE._serialized_start=1100
324
+ _LISTINFERENCEJOBFILESRESPONSE._serialized_end=1266
325
+ _LISTFILEDETAILS._serialized_start=1268
326
+ _LISTFILEDETAILS._serialized_end=1355
327
+ _CREATEINFERENCETASKEXECUTORREQUEST._serialized_start=1358
328
+ _CREATEINFERENCETASKEXECUTORREQUEST._serialized_end=1598
329
+ _PREPAREINFERENCEJOBREQUEST._serialized_start=1601
330
+ _PREPAREINFERENCEJOBREQUEST._serialized_end=1738
331
+ _PREPAREINFERENCEJOBRESPONSE._serialized_start=1740
332
+ _PREPAREINFERENCEJOBRESPONSE._serialized_end=1810
333
+ _TASKEXECUTORCONFIGURATIONMESSAGE._serialized_start=1813
334
+ _TASKEXECUTORCONFIGURATIONMESSAGE._serialized_end=2402
335
+ _INFERENCEJOBCONFIGURATIONMESSAGE._serialized_start=2404
336
+ _INFERENCEJOBCONFIGURATIONMESSAGE._serialized_end=2517
337
+ _MODELIDENTIFIER._serialized_start=2519
338
+ _MODELIDENTIFIER._serialized_end=2615
339
+ _LISTFILESPRESTEPCONFIGURATION._serialized_start=2618
340
+ _LISTFILESPRESTEPCONFIGURATION._serialized_end=2751
341
+ _INFERENCETASKCONFIGURATIONMESSAGE._serialized_start=2754
342
+ _INFERENCETASKCONFIGURATIONMESSAGE._serialized_end=3382
343
+ _BATCHJOBPARAMETER._serialized_start=3384
344
+ _BATCHJOBPARAMETER._serialized_end=3431
345
+ _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_start=3433
346
+ _CLEANINFERENCETASKSEXECUTORSREQUEST._serialized_end=3496
347
+ _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_start=3498
348
+ _CLEANINFERENCETASKSEXECUTORSRESPONSE._serialized_end=3577
349
+ _CLEANINFERENCETASKEXECUTORREQUEST._serialized_start=3579
350
+ _CLEANINFERENCETASKEXECUTORREQUEST._serialized_end=3667
351
+ _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_start=3669
352
+ _CLEANINFERENCETASKEXECUTORRESPONSE._serialized_end=3746
353
+ _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_start=3748
354
+ _CREATEINFERENCETASKEXECUTORRESPONSE._serialized_end=3826
355
+ _STARTINFERENCEJOBWARMUPREQUEST._serialized_start=3829
356
+ _STARTINFERENCEJOBWARMUPREQUEST._serialized_end=4204
357
+ _STARTINFERENCEJOBWARMUPRESPONSE._serialized_start=4206
358
+ _STARTINFERENCEJOBWARMUPRESPONSE._serialized_end=4239
359
+ _CANCELINFERENCEJOBWARMUPREQUEST._serialized_start=4241
360
+ _CANCELINFERENCEJOBWARMUPREQUEST._serialized_end=4353
361
+ _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_start=4355
362
+ _CANCELINFERENCEJOBWARMUPRESPONSE._serialized_end=4389
363
+ _BATCHJOBRESOURCES._serialized_start=4392
364
+ _BATCHJOBRESOURCES._serialized_end=4605
365
+ _GPURESOURCES._serialized_start=4607
366
+ _GPURESOURCES._serialized_end=4698
368
367
  # @@protoc_insertion_point(module_scope)
@@ -8,7 +8,6 @@ import google.protobuf.descriptor
8
8
  import google.protobuf.internal.containers
9
9
  import google.protobuf.internal.enum_type_wrapper
10
10
  import google.protobuf.message
11
- import google.protobuf.wrappers_pb2
12
11
  import sys
13
12
  import typing
14
13
 
@@ -526,6 +525,8 @@ class ModelIdentifier(google.protobuf.message.Message):
526
525
  global___ModelIdentifier = ModelIdentifier
527
526
 
528
527
  class ListFilesPreStepConfiguration(google.protobuf.message.Message):
528
+ """will be deprecated soon"""
529
+
529
530
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
530
531
 
531
532
  SOURCE_PATH_FIELD_NUMBER: builtins.int
@@ -538,18 +539,16 @@ class ListFilesPreStepConfiguration(google.protobuf.message.Message):
538
539
  """Name for the file to save the list of files in the source path"""
539
540
  total_number_of_tasks: builtins.int
540
541
  """Total number of inference tasks, used to divide the files list into similarly sized groups"""
541
- @property
542
- def task_index(self) -> google.protobuf.wrappers_pb2.Int32Value:
543
- """Index of the specific task, used to determine on which files group to run inference on. value 0 is legit"""
542
+ task_index: builtins.int
543
+ """Index of the specific task, used to determine on which files group to run inference on"""
544
544
  def __init__(
545
545
  self,
546
546
  *,
547
547
  source_path: builtins.str = ...,
548
548
  list_files_file_name: builtins.str = ...,
549
549
  total_number_of_tasks: builtins.int = ...,
550
- task_index: google.protobuf.wrappers_pb2.Int32Value | None = ...,
550
+ task_index: builtins.int = ...,
551
551
  ) -> None: ...
552
- def HasField(self, field_name: typing_extensions.Literal["task_index", b"task_index"]) -> builtins.bool: ...
553
552
  def ClearField(self, field_name: typing_extensions.Literal["list_files_file_name", b"list_files_file_name", "source_path", b"source_path", "task_index", b"task_index", "total_number_of_tasks", b"total_number_of_tasks"]) -> None: ...
554
553
 
555
554
  global___ListFilesPreStepConfiguration = ListFilesPreStepConfiguration
@@ -1 +1 @@
1
- __version__ = "0.0.103"
1
+ __version__ = "0.0.104"