luminarycloud 0.19.0__py3-none-any.whl → 0.20.0__py3-none-any.whl

luminarycloud/_client/client.py

@@ -129,6 +129,8 @@ class Client(
             ("grpc.keepalive_timeout_ms", 5000),
             ("grpc.keepalive_permit_without_calls", 1),
             ("grpc.http2.max_pings_without_data", 10),
+            ("grpc.max_receive_message_length", 32 * 1024 * 1024),
+            ("grpc.max_send_message_length", 32 * 1024 * 1024),
         ]
         if grpc_channel_options:
             grpc_channel_options_with_keep_alive.extend(grpc_channel_options)

luminarycloud/_helpers/_wait_for_mesh.py

@@ -1,7 +1,7 @@
 # Copyright 2023-2024 Luminary Cloud, Inc. All Rights Reserved.
 import logging
 from time import time, sleep
-from luminarycloud.exceptions import DeadlineExceededError
+from luminarycloud.exceptions import DeadlineExceededError, Timeout
 
 from .._proto.api.v0.luminarycloud.mesh.mesh_pb2 import Mesh, GetMeshRequest
 from .._client import Client
@@ -32,19 +32,20 @@ def wait_for_mesh(
 
     Raises
     ------
-    TimeoutError
+    Timeout
     """
     deadline = time() + timeout_seconds
     while True:
         if time() >= deadline:
-            logger.error("`wait_for_mesh` timed out.")
-            raise TimeoutError
+            raise Timeout("Timed out waiting for mesh to be ready.")
         # It seems this call sometimes hangs. It seems as well that python gRPC is known to hang
         # in some cases, so we'll add a deadline and catch deadline exceeded errors to retry.
         try:
             response = client.GetMesh(GetMeshRequest(id=mesh.id), timeout=15)
         except DeadlineExceededError:
-            logger.error("Deadline exceeded while waiting for mesh.")
+            logger.debug(
+                "Request deadline exceeded while waiting for mesh status. Will retry if the `wait_for_mesh` deadline has not expired."
+            )
             sleep(max(0, min(interval_seconds, deadline - time())))
             continue
         status = response.mesh.status

luminarycloud/_helpers/_wait_for_simulation.py

@@ -2,6 +2,7 @@
 import io
 import logging
 from time import time, sleep
+from luminarycloud.exceptions import Timeout
 
 from .._proto.api.v0.luminarycloud.simulation.simulation_pb2 import (
     GetSimulationResponse,
@@ -42,7 +43,7 @@ def wait_for_simulation(
 
     Raises
     ------
-    TimeoutError
+    Timeout
     """
     deadline = time() + timeout_seconds
     # latest_iter is used to avoid printing the same iteration's residuals multiple times
@@ -66,8 +67,7 @@ def wait_for_simulation(
         ]:
             return status
         if time() >= deadline:
-            logger.error("`wait_for_simulation` timed out.")
-            raise TimeoutError
+            raise Timeout("Timed out waiting for simulation to finish")
         sleep(max(0, min(interval_seconds, deadline - time())))
 
 

luminarycloud/_proto/api/v0/luminarycloud/physics_ai/physics_ai_pb2.py

@@ -16,18 +16,30 @@ _sym_db = _symbol_database.Default()
 from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 from luminarycloud._proto.api.v0.luminarycloud.common import common_pb2 as proto_dot_api_dot_v0_dot_luminarycloud_dot_common_dot_common__pb2
+from luminarycloud._proto.base import base_pb2 as proto_dot_base_dot_base__pb2
 from luminarycloud._proto.quantity import quantity_pb2 as proto_dot_quantity_dot_quantity__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6proto/api/v0/luminarycloud/physics_ai/physics_ai.proto\x12.luminary.proto.api.v0.luminarycloud.physics_ai\x1a\x1cgoogle/api/annotations.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a.proto/api/v0/luminarycloud/common/common.proto\x1a\x1dproto/quantity/quantity.proto\"\xad\x01\n\x1cPhysicsAiArchitectureVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x11\n\tchangelog\x18\x03 \x01(\t\x12`\n\x0flifecycle_state\x18\x04 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\xa6\x01\n\x15PhysicsAiArchitecture\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12^\n\x08versions\x18\x04 \x03(\x0b\x32L.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitectureVersion\"\x1a\n\x18ListArchitecturesRequest\"y\n\x19ListArchitecturesResponse\x12\\\n\rarchitectures\x18\x01 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitecture\"\x93\x01\n\x15PhysicsAiModelVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12`\n\x0flifecycle_state\x18\x03 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\x98\x01\n\x0ePhysicsAiModel\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12W\n\x08versions\x18\x04 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModelVersion\"\x1d\n\x1bListPretrainedModelsRequest\"n\n\x1cListPretrainedModelsResponse\x12N\n\x06models\x18\x01 \x03(\x0b\x32>.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModel\"\xa3\x02\n\x1fGetSolutionDataPhysicsAIRequest\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x18\n\x10\x65xclude_surfaces\x18\x02 \x03(\t\x12\x12\n\nfill_holes\x18\x03 \x01(\x02\x12\x45\n\x16surface_fields_to_keep\x18\x04 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x44\n\x15volume_fields_to_keep\x18\x05 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x16\n\x0eprocess_volume\x18\x06 \x01(\x08\x12\x18\n\x10single_precision\x18\x07 \x01(\x08\"b\n GetSolutionDataPhysicsAIResponse\x12>\n\x04\x66ile\x18\x01 \x01(\x0b\x32\x30.luminary.proto.api.v0.luminarycloud.common.File*\xb4\x01\n\x17PhysicsAiLifecycleState\x12\x1f\n\x1bLIFECYCLE_STATE_UNSPECIFIED\x10\x00\x12\x1f\n\x1bLIFECYCLE_STATE_DEVELOPMENT\x10\x01\x12\x1a\n\x16LIFECYCLE_STATE_ACTIVE\x10\x02\x12\x1e\n\x1aLIFECYCLE_STATE_DEPRECATED\x10\x03\x12\x1b\n\x17LIFECYCLE_STATE_RETIRED\x10\x04\x32\xb9\x05\n\x10PhysicsAiService\x12\xce\x01\n\x11ListArchitectures\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesResponse\"$\x82\xd3\xe4\x93\x02\x1e\x12\x1c/v0/physics_ai/architectures\x12\xdb\x01\n\x14ListPretrainedModels\x12K.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsRequest\x1aL.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsResponse\"(\x82\xd3\xe4\x93\x02\"\x12 /v0/physics_ai/pretrained_models\x12\xf5\x01\n\x18GetSolutionDataPhysicsAI\x12O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\x1aP.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIResponse\"6\x82\xd3\xe4\x93\x02\x30\"+/v0/physics_ai/solutions/{solution_id}/data:\x01*B>Z<luminarycloud.com/core/proto/api/v0/luminarycloud/physics_aib\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6proto/api/v0/luminarycloud/physics_ai/physics_ai.proto\x12.luminary.proto.api.v0.luminarycloud.physics_ai\x1a\x1cgoogle/api/annotations.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a.proto/api/v0/luminarycloud/common/common.proto\x1a\x15proto/base/base.proto\x1a\x1dproto/quantity/quantity.proto\"\xad\x01\n\x1cPhysicsAiArchitectureVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x11\n\tchangelog\x18\x03 \x01(\t\x12`\n\x0flifecycle_state\x18\x04 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\xa6\x01\n\x15PhysicsAiArchitecture\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12^\n\x08versions\x18\x04 \x03(\x0b\x32L.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitectureVersion\"\x1a\n\x18ListArchitecturesRequest\"y\n\x19ListArchitecturesResponse\x12\\\n\rarchitectures\x18\x01 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitecture\"\x93\x01\n\x15PhysicsAiModelVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12`\n\x0flifecycle_state\x18\x03 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\x98\x01\n\x0ePhysicsAiModel\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12W\n\x08versions\x18\x04 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModelVersion\"\x1d\n\x1bListPretrainedModelsRequest\"n\n\x1cListPretrainedModelsResponse\x12N\n\x06models\x18\x01 \x03(\x0b\x32>.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModel\"\xa3\x02\n\x1fGetSolutionDataPhysicsAIRequest\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x18\n\x10\x65xclude_surfaces\x18\x02 \x03(\t\x12\x12\n\nfill_holes\x18\x03 \x01(\x02\x12\x45\n\x16surface_fields_to_keep\x18\x04 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x44\n\x15volume_fields_to_keep\x18\x05 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x16\n\x0eprocess_volume\x18\x06 \x01(\x08\x12\x18\n\x10single_precision\x18\x07 \x01(\x08\"b\n GetSolutionDataPhysicsAIResponse\x12>\n\x04\x66ile\x18\x01 \x01(\x0b\x32\x30.luminary.proto.api.v0.luminarycloud.common.File\";\n\x10TrainingSolution\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x12\n\ndata_split\x18\x02 \x01(\t\"\x9a\x05\n\x14PhysicsAiTrainingJob\x12\n\n\x02id\x18\x01 \x01(\t\x12\x1f\n\x17\x61rchitecture_version_id\x18\x02 \x01(\t\x12\x0f\n\x07user_id\x18\x03 \x01(\t\x12\x17\n\x0ftraining_config\x18\x04 \x01(\t\x12i\n\x19training_data_source_type\x18\x05 \x01(\x0e\x32\x46.luminary.proto.api.v0.luminarycloud.physics_ai.TrainingDataSourceType\x12\x1c\n\x14training_description\x18\x06 \x01(\t\x12\x1c\n\x14\x65xternal_dataset_uri\x18\x07 \x01(\t\x12\x64\n\x13initialization_type\x18\x08 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.ModelInitializationType\x12\x1d\n\x15\x62\x61se_model_version_id\x18\t \x01(\t\x12.\n\x06status\x18\n \x01(\x0b\x32\x1e.luminary.proto.base.JobStatus\x12\x15\n\rerror_message\x18\x0b \x01(\t\x12\x1f\n\x17output_model_version_id\x18\x0c \x01(\t\x12\x31\n\rcreation_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0bupdate_time\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x33\n\x0f\x63ompletion_time\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xf3\x02\n\x18SubmitTrainingJobRequest\x12\x1f\n\x17\x61rchitecture_version_id\x18\x01 \x01(\t\x12\x1c\n\x14training_description\x18\x02 \x01(\t\x12\x1c\n\x14\x65xternal_dataset_uri\x18\x03 \x01(\t\x12\\\n\x12training_solutions\x18\x04 \x03(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.TrainingSolution\x12\x17\n\x0ftraining_config\x18\x05 \x01(\t\x12\x64\n\x13initialization_type\x18\x06 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.ModelInitializationType\x12\x1d\n\x15\x62\x61se_model_version_id\x18\x07 \x01(\t\"w\n\x19SubmitTrainingJobResponse\x12Z\n\x0ctraining_job\x18\x01 \x01(\x0b\x32\x44.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiTrainingJob*\xb4\x01\n\x17PhysicsAiLifecycleState\x12\x1f\n\x1bLIFECYCLE_STATE_UNSPECIFIED\x10\x00\x12\x1f\n\x1bLIFECYCLE_STATE_DEVELOPMENT\x10\x01\x12\x1a\n\x16LIFECYCLE_STATE_ACTIVE\x10\x02\x12\x1e\n\x1aLIFECYCLE_STATE_DEPRECATED\x10\x03\x12\x1b\n\x17LIFECYCLE_STATE_RETIRED\x10\x04*\xa7\x01\n\x16TrainingDataSourceType\x12)\n%TRAINING_DATA_SOURCE_TYPE_UNSPECIFIED\x10\x00\x12\x32\n.TRAINING_DATA_SOURCE_TYPE_SIMULATION_SOLUTIONS\x10\x01\x12.\n*TRAINING_DATA_SOURCE_TYPE_EXTERNAL_DATASET\x10\x02*\xc1\x01\n\x17ModelInitializationType\x12)\n%MODEL_INITIALIZATION_TYPE_UNSPECIFIED\x10\x00\x12$\n MODEL_INITIALIZATION_TYPE_RANDOM\x10\x01\x12+\n\'MODEL_INITIALIZATION_TYPE_MODEL_VERSION\x10\x02\x12(\n$MODEL_INITIALIZATION_TYPE_CHECKPOINT\x10\x03\x32\x8d\x07\n\x10PhysicsAiService\x12\xce\x01\n\x11ListArchitectures\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesResponse\"$\x82\xd3\xe4\x93\x02\x1e\x12\x1c/v0/physics_ai/architectures\x12\xdb\x01\n\x14ListPretrainedModels\x12K.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsRequest\x1aL.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsResponse\"(\x82\xd3\xe4\x93\x02\"\x12 /v0/physics_ai/pretrained_models\x12\xf5\x01\n\x18GetSolutionDataPhysicsAI\x12O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\x1aP.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIResponse\"6\x82\xd3\xe4\x93\x02\x30\"+/v0/physics_ai/solutions/{solution_id}/data:\x01*\x12\xd1\x01\n\x11SubmitTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobResponse\"\'\x82\xd3\xe4\x93\x02!\"\x1c/v0/physics_ai/training/jobs:\x01*B>Z<luminarycloud.com/core/proto/api/v0/luminarycloud/physics_aib\x06proto3')
 
 _PHYSICSAILIFECYCLESTATE = DESCRIPTOR.enum_types_by_name['PhysicsAiLifecycleState']
 PhysicsAiLifecycleState = enum_type_wrapper.EnumTypeWrapper(_PHYSICSAILIFECYCLESTATE)
+_TRAININGDATASOURCETYPE = DESCRIPTOR.enum_types_by_name['TrainingDataSourceType']
+TrainingDataSourceType = enum_type_wrapper.EnumTypeWrapper(_TRAININGDATASOURCETYPE)
+_MODELINITIALIZATIONTYPE = DESCRIPTOR.enum_types_by_name['ModelInitializationType']
+ModelInitializationType = enum_type_wrapper.EnumTypeWrapper(_MODELINITIALIZATIONTYPE)
 LIFECYCLE_STATE_UNSPECIFIED = 0
 LIFECYCLE_STATE_DEVELOPMENT = 1
 LIFECYCLE_STATE_ACTIVE = 2
 LIFECYCLE_STATE_DEPRECATED = 3
 LIFECYCLE_STATE_RETIRED = 4
+TRAINING_DATA_SOURCE_TYPE_UNSPECIFIED = 0
+TRAINING_DATA_SOURCE_TYPE_SIMULATION_SOLUTIONS = 1
+TRAINING_DATA_SOURCE_TYPE_EXTERNAL_DATASET = 2
+MODEL_INITIALIZATION_TYPE_UNSPECIFIED = 0
+MODEL_INITIALIZATION_TYPE_RANDOM = 1
+MODEL_INITIALIZATION_TYPE_MODEL_VERSION = 2
+MODEL_INITIALIZATION_TYPE_CHECKPOINT = 3
 
 
 _PHYSICSAIARCHITECTUREVERSION = DESCRIPTOR.message_types_by_name['PhysicsAiArchitectureVersion']
@@ -40,6 +52,10 @@ _LISTPRETRAINEDMODELSREQUEST = DESCRIPTOR.message_types_by_name['ListPretrainedM
 _LISTPRETRAINEDMODELSRESPONSE = DESCRIPTOR.message_types_by_name['ListPretrainedModelsResponse']
 _GETSOLUTIONDATAPHYSICSAIREQUEST = DESCRIPTOR.message_types_by_name['GetSolutionDataPhysicsAIRequest']
 _GETSOLUTIONDATAPHYSICSAIRESPONSE = DESCRIPTOR.message_types_by_name['GetSolutionDataPhysicsAIResponse']
+_TRAININGSOLUTION = DESCRIPTOR.message_types_by_name['TrainingSolution']
+_PHYSICSAITRAININGJOB = DESCRIPTOR.message_types_by_name['PhysicsAiTrainingJob']
+_SUBMITTRAININGJOBREQUEST = DESCRIPTOR.message_types_by_name['SubmitTrainingJobRequest']
+_SUBMITTRAININGJOBRESPONSE = DESCRIPTOR.message_types_by_name['SubmitTrainingJobResponse']
 PhysicsAiArchitectureVersion = _reflection.GeneratedProtocolMessageType('PhysicsAiArchitectureVersion', (_message.Message,), {
   'DESCRIPTOR' : _PHYSICSAIARCHITECTUREVERSION,
   '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
@@ -110,6 +126,34 @@ GetSolutionDataPhysicsAIResponse = _reflection.GeneratedProtocolMessageType('Get
   })
 _sym_db.RegisterMessage(GetSolutionDataPhysicsAIResponse)
 
+TrainingSolution = _reflection.GeneratedProtocolMessageType('TrainingSolution', (_message.Message,), {
+  'DESCRIPTOR' : _TRAININGSOLUTION,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.TrainingSolution)
+  })
+_sym_db.RegisterMessage(TrainingSolution)
+
+PhysicsAiTrainingJob = _reflection.GeneratedProtocolMessageType('PhysicsAiTrainingJob', (_message.Message,), {
+  'DESCRIPTOR' : _PHYSICSAITRAININGJOB,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiTrainingJob)
+  })
+_sym_db.RegisterMessage(PhysicsAiTrainingJob)
+
+SubmitTrainingJobRequest = _reflection.GeneratedProtocolMessageType('SubmitTrainingJobRequest', (_message.Message,), {
+  'DESCRIPTOR' : _SUBMITTRAININGJOBREQUEST,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobRequest)
+  })
+_sym_db.RegisterMessage(SubmitTrainingJobRequest)
+
+SubmitTrainingJobResponse = _reflection.GeneratedProtocolMessageType('SubmitTrainingJobResponse', (_message.Message,), {
+  'DESCRIPTOR' : _SUBMITTRAININGJOBRESPONSE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobResponse)
+  })
+_sym_db.RegisterMessage(SubmitTrainingJobResponse)
+
 _PHYSICSAISERVICE = DESCRIPTOR.services_by_name['PhysicsAiService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -121,28 +165,42 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PHYSICSAISERVICE.methods_by_name['ListPretrainedModels']._serialized_options = b'\202\323\344\223\002\"\022 /v0/physics_ai/pretrained_models'
   _PHYSICSAISERVICE.methods_by_name['GetSolutionDataPhysicsAI']._options = None
   _PHYSICSAISERVICE.methods_by_name['GetSolutionDataPhysicsAI']._serialized_options = b'\202\323\344\223\0020\"+/v0/physics_ai/solutions/{solution_id}/data:\001*'
-  _PHYSICSAILIFECYCLESTATE._serialized_start=1587
-  _PHYSICSAILIFECYCLESTATE._serialized_end=1767
-  _PHYSICSAIARCHITECTUREVERSION._serialized_start=249
-  _PHYSICSAIARCHITECTUREVERSION._serialized_end=422
-  _PHYSICSAIARCHITECTURE._serialized_start=425
-  _PHYSICSAIARCHITECTURE._serialized_end=591
-  _LISTARCHITECTURESREQUEST._serialized_start=593
-  _LISTARCHITECTURESREQUEST._serialized_end=619
-  _LISTARCHITECTURESRESPONSE._serialized_start=621
-  _LISTARCHITECTURESRESPONSE._serialized_end=742
-  _PHYSICSAIMODELVERSION._serialized_start=745
-  _PHYSICSAIMODELVERSION._serialized_end=892
-  _PHYSICSAIMODEL._serialized_start=895
-  _PHYSICSAIMODEL._serialized_end=1047
-  _LISTPRETRAINEDMODELSREQUEST._serialized_start=1049
-  _LISTPRETRAINEDMODELSREQUEST._serialized_end=1078
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1080
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1190
-  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_start=1193
-  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_end=1484
-  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_start=1486
-  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_end=1584
-  _PHYSICSAISERVICE._serialized_start=1770
-  _PHYSICSAISERVICE._serialized_end=2467
+  _PHYSICSAISERVICE.methods_by_name['SubmitTrainingJob']._options = None
+  _PHYSICSAISERVICE.methods_by_name['SubmitTrainingJob']._serialized_options = b'\202\323\344\223\002!\"\034/v0/physics_ai/training/jobs:\001*'
+  _PHYSICSAILIFECYCLESTATE._serialized_start=2835
+  _PHYSICSAILIFECYCLESTATE._serialized_end=3015
+  _TRAININGDATASOURCETYPE._serialized_start=3018
+  _TRAININGDATASOURCETYPE._serialized_end=3185
+  _MODELINITIALIZATIONTYPE._serialized_start=3188
+  _MODELINITIALIZATIONTYPE._serialized_end=3381
+  _PHYSICSAIARCHITECTUREVERSION._serialized_start=272
+  _PHYSICSAIARCHITECTUREVERSION._serialized_end=445
+  _PHYSICSAIARCHITECTURE._serialized_start=448
+  _PHYSICSAIARCHITECTURE._serialized_end=614
+  _LISTARCHITECTURESREQUEST._serialized_start=616
+  _LISTARCHITECTURESREQUEST._serialized_end=642
+  _LISTARCHITECTURESRESPONSE._serialized_start=644
+  _LISTARCHITECTURESRESPONSE._serialized_end=765
+  _PHYSICSAIMODELVERSION._serialized_start=768
+  _PHYSICSAIMODELVERSION._serialized_end=915
+  _PHYSICSAIMODEL._serialized_start=918
+  _PHYSICSAIMODEL._serialized_end=1070
+  _LISTPRETRAINEDMODELSREQUEST._serialized_start=1072
+  _LISTPRETRAINEDMODELSREQUEST._serialized_end=1101
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1103
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1213
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_start=1216
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_end=1507
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_start=1509
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_end=1607
+  _TRAININGSOLUTION._serialized_start=1609
+  _TRAININGSOLUTION._serialized_end=1668
+  _PHYSICSAITRAININGJOB._serialized_start=1671
+  _PHYSICSAITRAININGJOB._serialized_end=2337
+  _SUBMITTRAININGJOBREQUEST._serialized_start=2340
+  _SUBMITTRAININGJOBREQUEST._serialized_end=2711
+  _SUBMITTRAININGJOBRESPONSE._serialized_start=2713
+  _SUBMITTRAININGJOBRESPONSE._serialized_end=2832
+  _PHYSICSAISERVICE._serialized_start=3384
+  _PHYSICSAISERVICE._serialized_end=4293
 # @@protoc_insertion_point(module_scope)

luminarycloud/_proto/api/v0/luminarycloud/physics_ai/physics_ai_pb2.pyi

@@ -8,7 +8,9 @@ import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
+import google.protobuf.timestamp_pb2
 import luminarycloud._proto.api.v0.luminarycloud.common.common_pb2
+import luminarycloud._proto.base.base_pb2
 import luminarycloud._proto.quantity.quantity_pb2
 import sys
 import typing
@@ -52,6 +54,58 @@ LIFECYCLE_STATE_RETIRED: PhysicsAiLifecycleState.ValueType  # 4
 """Archived; no training or inference support."""
 global___PhysicsAiLifecycleState = PhysicsAiLifecycleState
 
+class _TrainingDataSourceType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _TrainingDataSourceTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TrainingDataSourceType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    TRAINING_DATA_SOURCE_TYPE_UNSPECIFIED: _TrainingDataSourceType.ValueType  # 0
+    """Default value, should not be used in practice."""
+    TRAINING_DATA_SOURCE_TYPE_SIMULATION_SOLUTIONS: _TrainingDataSourceType.ValueType  # 1
+    """Training data from simulation solutions."""
+    TRAINING_DATA_SOURCE_TYPE_EXTERNAL_DATASET: _TrainingDataSourceType.ValueType  # 2
+    """Training data from external dataset."""
+
+class TrainingDataSourceType(_TrainingDataSourceType, metaclass=_TrainingDataSourceTypeEnumTypeWrapper):
+    """Training data source types"""
+
+TRAINING_DATA_SOURCE_TYPE_UNSPECIFIED: TrainingDataSourceType.ValueType  # 0
+"""Default value, should not be used in practice."""
+TRAINING_DATA_SOURCE_TYPE_SIMULATION_SOLUTIONS: TrainingDataSourceType.ValueType  # 1
+"""Training data from simulation solutions."""
+TRAINING_DATA_SOURCE_TYPE_EXTERNAL_DATASET: TrainingDataSourceType.ValueType  # 2
+"""Training data from external dataset."""
+global___TrainingDataSourceType = TrainingDataSourceType
+
+class _ModelInitializationType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _ModelInitializationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ModelInitializationType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    MODEL_INITIALIZATION_TYPE_UNSPECIFIED: _ModelInitializationType.ValueType  # 0
+    """Default value, should not be used in practice."""
+    MODEL_INITIALIZATION_TYPE_RANDOM: _ModelInitializationType.ValueType  # 1
+    """Random initialization."""
+    MODEL_INITIALIZATION_TYPE_MODEL_VERSION: _ModelInitializationType.ValueType  # 2
+    """Initialize from existing model version."""
+    MODEL_INITIALIZATION_TYPE_CHECKPOINT: _ModelInitializationType.ValueType  # 3
+    """Initialize from checkpoint."""
+
+class ModelInitializationType(_ModelInitializationType, metaclass=_ModelInitializationTypeEnumTypeWrapper):
+    """Model initialization types"""
+
+MODEL_INITIALIZATION_TYPE_UNSPECIFIED: ModelInitializationType.ValueType  # 0
+"""Default value, should not be used in practice."""
+MODEL_INITIALIZATION_TYPE_RANDOM: ModelInitializationType.ValueType  # 1
+"""Random initialization."""
+MODEL_INITIALIZATION_TYPE_MODEL_VERSION: ModelInitializationType.ValueType  # 2
+"""Initialize from existing model version."""
+MODEL_INITIALIZATION_TYPE_CHECKPOINT: ModelInitializationType.ValueType  # 3
+"""Initialize from checkpoint."""
+global___ModelInitializationType = ModelInitializationType
+
 class PhysicsAiArchitectureVersion(google.protobuf.message.Message):
     """A physics ai architecture version."""
 
@@ -290,3 +344,163 @@ class GetSolutionDataPhysicsAIResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["file", b"file"]) -> None: ...
 
 global___GetSolutionDataPhysicsAIResponse = GetSolutionDataPhysicsAIResponse
+
+class TrainingSolution(google.protobuf.message.Message):
+    """A training solution specification"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOLUTION_ID_FIELD_NUMBER: builtins.int
+    DATA_SPLIT_FIELD_NUMBER: builtins.int
+    solution_id: builtins.str
+    """Solution ID to use for training."""
+    data_split: builtins.str
+    """Data split for this solution (train, validation, test)."""
+    def __init__(
+        self,
+        *,
+        solution_id: builtins.str = ...,
+        data_split: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data_split", b"data_split", "solution_id", b"solution_id"]) -> None: ...
+
+global___TrainingSolution = TrainingSolution
+
+class PhysicsAiTrainingJob(google.protobuf.message.Message):
+    """A Physics AI training job"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    ARCHITECTURE_VERSION_ID_FIELD_NUMBER: builtins.int
+    USER_ID_FIELD_NUMBER: builtins.int
+    TRAINING_CONFIG_FIELD_NUMBER: builtins.int
+    TRAINING_DATA_SOURCE_TYPE_FIELD_NUMBER: builtins.int
+    TRAINING_DESCRIPTION_FIELD_NUMBER: builtins.int
+    EXTERNAL_DATASET_URI_FIELD_NUMBER: builtins.int
+    INITIALIZATION_TYPE_FIELD_NUMBER: builtins.int
+    BASE_MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    ERROR_MESSAGE_FIELD_NUMBER: builtins.int
+    OUTPUT_MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    CREATION_TIME_FIELD_NUMBER: builtins.int
+    UPDATE_TIME_FIELD_NUMBER: builtins.int
+    COMPLETION_TIME_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    """Unique identifier for the training job."""
+    architecture_version_id: builtins.str
+    """Architecture version ID being trained."""
+    user_id: builtins.str
+    """User ID who submitted the job."""
+    training_config: builtins.str
+    """Training configuration as JSON."""
+    training_data_source_type: global___TrainingDataSourceType.ValueType
+    """Type of training data source."""
+    training_description: builtins.str
+    """Description of the training job. (mutable)"""
+    external_dataset_uri: builtins.str
+    """External dataset URI (if using external dataset)."""
+    initialization_type: global___ModelInitializationType.ValueType
+    """Model initialization type."""
+    base_model_version_id: builtins.str
+    """Base model version ID (if initializing from existing model)."""
+    @property
+    def status(self) -> luminarycloud._proto.base.base_pb2.JobStatus:
+        """Current status of the training job. (mutable)"""
+    error_message: builtins.str
+    """Error message (if failed)."""
+    output_model_version_id: builtins.str
+    """Output model version ID (if completed, so mutable)."""
+    @property
+    def creation_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Job creation time."""
+    @property
+    def update_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Job last update time, updated when status changes. (mutable)"""
+    @property
+    def completion_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Job completion time, set upon job completion. (mutable)"""
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+        architecture_version_id: builtins.str = ...,
+        user_id: builtins.str = ...,
+        training_config: builtins.str = ...,
+        training_data_source_type: global___TrainingDataSourceType.ValueType = ...,
+        training_description: builtins.str = ...,
+        external_dataset_uri: builtins.str = ...,
+        initialization_type: global___ModelInitializationType.ValueType = ...,
+        base_model_version_id: builtins.str = ...,
+        status: luminarycloud._proto.base.base_pb2.JobStatus | None = ...,
+        error_message: builtins.str = ...,
+        output_model_version_id: builtins.str = ...,
+        creation_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        update_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        completion_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["completion_time", b"completion_time", "creation_time", b"creation_time", "status", b"status", "update_time", b"update_time"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["architecture_version_id", b"architecture_version_id", "base_model_version_id", b"base_model_version_id", "completion_time", b"completion_time", "creation_time", b"creation_time", "error_message", b"error_message", "external_dataset_uri", b"external_dataset_uri", "id", b"id", "initialization_type", b"initialization_type", "output_model_version_id", b"output_model_version_id", "status", b"status", "training_config", b"training_config", "training_data_source_type", b"training_data_source_type", "training_description", b"training_description", "update_time", b"update_time", "user_id", b"user_id"]) -> None: ...
+
+global___PhysicsAiTrainingJob = PhysicsAiTrainingJob
+
+class SubmitTrainingJobRequest(google.protobuf.message.Message):
+    """Request message for submitting a training job"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ARCHITECTURE_VERSION_ID_FIELD_NUMBER: builtins.int
+    TRAINING_DESCRIPTION_FIELD_NUMBER: builtins.int
+    EXTERNAL_DATASET_URI_FIELD_NUMBER: builtins.int
+    TRAINING_SOLUTIONS_FIELD_NUMBER: builtins.int
+    TRAINING_CONFIG_FIELD_NUMBER: builtins.int
+    INITIALIZATION_TYPE_FIELD_NUMBER: builtins.int
+    BASE_MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    architecture_version_id: builtins.str
+    """Architecture version ID to train."""
+    training_description: builtins.str
+    """Description of the training job."""
+    external_dataset_uri: builtins.str
+    """External dataset URI"""
+    @property
+    def training_solutions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrainingSolution]:
+        """Training solutions"""
+    training_config: builtins.str
+    """Training configuration in JSON format."""
+    initialization_type: global___ModelInitializationType.ValueType
+    """Model initialization type."""
+    base_model_version_id: builtins.str
+    """Base model version ID (if initializing from existing model)."""
+    def __init__(
+        self,
+        *,
+        architecture_version_id: builtins.str = ...,
+        training_description: builtins.str = ...,
+        external_dataset_uri: builtins.str = ...,
+        training_solutions: collections.abc.Iterable[global___TrainingSolution] | None = ...,
+        training_config: builtins.str = ...,
+        initialization_type: global___ModelInitializationType.ValueType = ...,
+        base_model_version_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["architecture_version_id", b"architecture_version_id", "base_model_version_id", b"base_model_version_id", "external_dataset_uri", b"external_dataset_uri", "initialization_type", b"initialization_type", "training_config", b"training_config", "training_description", b"training_description", "training_solutions", b"training_solutions"]) -> None: ...
+
+global___SubmitTrainingJobRequest = SubmitTrainingJobRequest
+
+class SubmitTrainingJobResponse(google.protobuf.message.Message):
+    """Response message for submitting a training job"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOB_FIELD_NUMBER: builtins.int
+    @property
+    def training_job(self) -> global___PhysicsAiTrainingJob:
+        """The created training job."""
+    def __init__(
+        self,
+        *,
+        training_job: global___PhysicsAiTrainingJob | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["training_job", b"training_job"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["training_job", b"training_job"]) -> None: ...
+
+global___SubmitTrainingJobResponse = SubmitTrainingJobResponse

luminarycloud/_proto/api/v0/luminarycloud/physics_ai/physics_ai_pb2_grpc.py

@@ -30,6 +30,11 @@ class PhysicsAiServiceStub(object):
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIRequest.SerializeToString,
                 response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIResponse.FromString,
                 )
+        self.SubmitTrainingJob = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/SubmitTrainingJob',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobResponse.FromString,
+                )
 
 
 class PhysicsAiServiceServicer(object):
@@ -57,6 +62,13 @@ class PhysicsAiServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def SubmitTrainingJob(self, request, context):
+        """Submits a training job for a Physics AI architecture
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_PhysicsAiServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -75,6 +87,11 @@ def add_PhysicsAiServiceServicer_to_server(servicer, server):
                     request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIRequest.FromString,
                     response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIResponse.SerializeToString,
             ),
+            'SubmitTrainingJob': grpc.unary_unary_rpc_method_handler(
+                    servicer.SubmitTrainingJob,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService', rpc_method_handlers)
@@ -136,3 +153,20 @@ class PhysicsAiService(object):
             proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def SubmitTrainingJob(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, '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/SubmitTrainingJob',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

luminarycloud/_proto/api/v0/luminarycloud/physics_ai/physics_ai_pb2_grpc.pyi

@@ -25,6 +25,11 @@ class PhysicsAiServiceStub:
         luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetSolutionDataPhysicsAIResponse,
     ]
     """Gets solution data with physics AI processing applied"""
+    SubmitTrainingJob: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobRequest,
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse,
+    ]
+    """Submits a training job for a Physics AI architecture"""
 
 class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
     """Manages physics ai architectures."""
@@ -50,5 +55,12 @@ class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetSolutionDataPhysicsAIResponse:
         """Gets solution data with physics AI processing applied"""
+    @abc.abstractmethod
+    def SubmitTrainingJob(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse:
+        """Submits a training job for a Physics AI architecture"""
 
 def add_PhysicsAiServiceServicer_to_server(servicer: PhysicsAiServiceServicer, server: grpc.Server) -> None: ...