luminarycloud 0.22.3__py3-none-any.whl → 0.23.1__py3-none-any.whl

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

@@ -21,7 +21,7 @@ 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/protobuf/struct.proto\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\".\n\x0cSurfaceGroup\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08surfaces\x18\x02 \x03(\t\"\xe9\x04\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\x12~\n\x10internal_options\x18\x08 \x03(\x0b\x32\x64.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest.InternalOptionsEntry\x12[\n\x15\x65xport_surface_groups\x18\t \x03(\x0b\x32<.luminary.proto.api.v0.luminarycloud.physics_ai.SurfaceGroup\x12\x19\n\x11skip_tar_creation\x18\n \x01(\x08\x12\x14\n\x0c\x65xclude_tags\x18\x0b \x03(\t\x1a\x36\n\x14InternalOptionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xe6\x01\n GetSolutionDataPhysicsAIResponse\x12>\n\x04\x66ile\x18\x01 \x01(\x0b\x32\x30.luminary.proto.api.v0.luminarycloud.common.File\x12\x63\n\x10physics_ai_files\x18\x02 \x01(\x0b\x32I.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFiles\x12\x1d\n\x15physics_ai_folder_url\x18\x03 \x01(\t\";\n\x10TrainingSolution\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x12\n\ndata_split\x18\x02 \x01(\t\"\xae\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\x12\x12\n\ndataset_id\x18\x10 \x01(\t\"\x87\x03\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\x12\x12\n\ndataset_id\x18\x08 \x01(\t\"w\n\x19SubmitTrainingJobResponse\x12Z\n\x0ctraining_job\x18\x01 \x01(\x0b\x32\x44.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiTrainingJob\"3\n\x18\x43\x61ncelTrainingJobRequest\x12\x17\n\x0ftraining_job_id\x18\x01 \x01(\t\"\x1b\n\x19\x43\x61ncelTrainingJobResponse\"\x92\x03\n\x10PhysicsAiDataset\x12\n\n\x02id\x18\x01 \x01(\t\x12\x17\n\ncreated_by\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x18\n\x0b\x64\x65scription\x18\x04 \x01(\tH\x01\x88\x01\x01\x12\x11\n\tis_locked\x18\x05 \x01(\x08\x12\x31\n\rcreation_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0bupdate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12-\n\tlocked_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x66\n\rexport_config\x18\n \x01(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequestB\r\n\x0b_created_byB\x0e\n\x0c_descriptionJ\x04\x08\t\x10\n\";\n\x18PhysicsAiDatasetCaseFile\x12\x11\n\tfile_type\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\"t\n\x19PhysicsAiDatasetCaseFiles\x12W\n\x05\x66iles\x18\x01 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\"\x87\x02\n\x14PhysicsAiDatasetCase\x12\x0f\n\x07\x63\x61se_id\x18\x01 \x01(\t\x12\x13\n\x0bsolution_id\x18\x02 \x01(\t\x12\x15\n\rsimulation_id\x18\x03 \x01(\t\x12\'\n\x06params\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\x12W\n\x05\x66iles\x18\x05 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\x12\x30\n\x0fsource_metadata\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\"\x81\x02\n\x1f\x43reatePhysicsAiDatasetCaseInput\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x15\n\rsimulation_id\x18\x02 \x01(\t\x12\'\n\x06params\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12W\n\x05\x66iles\x18\x04 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\x12\x30\n\x0fsource_metadata\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xb4\x02\n\x14\x43reateDatasetRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12^\n\x05\x63\x61ses\x18\x03 \x03(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.CreatePhysicsAiDatasetCaseInput\x12\x31\n\x10parameter_schema\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x66\n\rexport_config\x18\x05 \x01(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\"j\n\x15\x43reateDatasetResponse\x12Q\n\x07\x64\x61taset\x18\x01 \x01(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDataset\"\x15\n\x13ListDatasetsRequest\"j\n\x14ListDatasetsResponse\x12R\n\x08\x64\x61tasets\x18\x01 \x03(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDataset*\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\n\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*\x12\xc0\x01\n\rCreateDataset\x12\x44.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetRequest\x1a\x45.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetResponse\"\"\x82\xd3\xe4\x93\x02\x1c\"\x17/v0/physics_ai/datasets:\x01*\x12\xba\x01\n\x0cListDatasets\x12\x43.luminary.proto.api.v0.luminarycloud.physics_ai.ListDatasetsRequest\x1a\x44.luminary.proto.api.v0.luminarycloud.physics_ai.ListDatasetsResponse\"\x1f\x82\xd3\xe4\x93\x02\x19\x12\x17/v0/physics_ai/datasetsB>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/protobuf/struct.proto\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\"t\n\x13ParameterDefinition\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t\x12\x11\n\tdimension\x18\x03 \x01(\x05\x12\x19\n\x11\x64\x65\x66\x61ult_reference\x18\x04 \x03(\x02\x12\x13\n\x0b\x64\x65scription\x18\x05 \x01(\t\"j\n\x0fParameterSchema\x12W\n\nparameters\x18\x01 \x03(\x0b\x32\x43.luminary.proto.api.v0.luminarycloud.physics_ai.ParameterDefinition\"\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\".\n\x0cSurfaceGroup\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08surfaces\x18\x02 \x03(\t\"\xe9\x04\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\x12~\n\x10internal_options\x18\x08 \x03(\x0b\x32\x64.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest.InternalOptionsEntry\x12[\n\x15\x65xport_surface_groups\x18\t \x03(\x0b\x32<.luminary.proto.api.v0.luminarycloud.physics_ai.SurfaceGroup\x12\x19\n\x11skip_tar_creation\x18\n \x01(\x08\x12\x14\n\x0c\x65xclude_tags\x18\x0b \x03(\t\x1a\x36\n\x14InternalOptionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xe6\x01\n GetSolutionDataPhysicsAIResponse\x12>\n\x04\x66ile\x18\x01 \x01(\x0b\x32\x30.luminary.proto.api.v0.luminarycloud.common.File\x12\x63\n\x10physics_ai_files\x18\x02 \x01(\x0b\x32I.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFiles\x12\x1d\n\x15physics_ai_folder_url\x18\x03 \x01(\t\";\n\x10TrainingSolution\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x12\n\ndata_split\x18\x02 \x01(\t\"\xae\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\x12\x12\n\ndataset_id\x18\x10 \x01(\t\"9\n\x16TrainingInputParameter\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\treference\x18\x02 \x03(\x02\"\xe8\x04\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\x12\x12\n\ndataset_id\x18\x08 \x01(\t\x12>\n\x0fsurface_outputs\x18\t \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12=\n\x0evolume_outputs\x18\n \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12`\n\x10input_parameters\x18\x0b \x03(\x0b\x32\x46.luminary.proto.api.v0.luminarycloud.physics_ai.TrainingInputParameter\"w\n\x19SubmitTrainingJobResponse\x12Z\n\x0ctraining_job\x18\x01 \x01(\x0b\x32\x44.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiTrainingJob\"3\n\x18\x43\x61ncelTrainingJobRequest\x12\x17\n\x0ftraining_job_id\x18\x01 \x01(\t\"\x1b\n\x19\x43\x61ncelTrainingJobResponse\"0\n\x15GetTrainingJobRequest\x12\x17\n\x0ftraining_job_id\x18\x01 \x01(\t\"t\n\x16GetTrainingJobResponse\x12Z\n\x0ctraining_job\x18\x01 \x01(\x0b\x32\x44.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiTrainingJob\"\x92\x03\n\x10PhysicsAiDataset\x12\n\n\x02id\x18\x01 \x01(\t\x12\x17\n\ncreated_by\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x18\n\x0b\x64\x65scription\x18\x04 \x01(\tH\x01\x88\x01\x01\x12\x11\n\tis_locked\x18\x05 \x01(\x08\x12\x31\n\rcreation_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0bupdate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12-\n\tlocked_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x66\n\rexport_config\x18\n \x01(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequestB\r\n\x0b_created_byB\x0e\n\x0c_descriptionJ\x04\x08\t\x10\n\";\n\x18PhysicsAiDatasetCaseFile\x12\x11\n\tfile_type\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\"t\n\x19PhysicsAiDatasetCaseFiles\x12W\n\x05\x66iles\x18\x01 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\"\x87\x02\n\x14PhysicsAiDatasetCase\x12\x0f\n\x07\x63\x61se_id\x18\x01 \x01(\t\x12\x13\n\x0bsolution_id\x18\x02 \x01(\t\x12\x15\n\rsimulation_id\x18\x03 \x01(\t\x12\'\n\x06params\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\x12W\n\x05\x66iles\x18\x05 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\x12\x30\n\x0fsource_metadata\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\"\x81\x02\n\x1f\x43reatePhysicsAiDatasetCaseInput\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x15\n\rsimulation_id\x18\x02 \x01(\t\x12\'\n\x06params\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12W\n\x05\x66iles\x18\x04 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\x12\x30\n\x0fsource_metadata\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xdc\x02\n\x14\x43reateDatasetRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12^\n\x05\x63\x61ses\x18\x03 \x03(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.CreatePhysicsAiDatasetCaseInput\x12Y\n\x10parameter_schema\x18\x04 \x01(\x0b\x32?.luminary.proto.api.v0.luminarycloud.physics_ai.ParameterSchema\x12\x66\n\rexport_config\x18\x05 \x01(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\"j\n\x15\x43reateDatasetResponse\x12Q\n\x07\x64\x61taset\x18\x01 \x01(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDataset\"\x15\n\x13ListDatasetsRequest\"j\n\x14ListDatasetsResponse\x12R\n\x08\x64\x61tasets\x18\x01 \x03(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDataset\"\x19\n\x17ListTrainingJobsRequest\"w\n\x18ListTrainingJobsResponse\x12[\n\rtraining_jobs\x18\x01 \x03(\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\xd4\r\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*\x12\xea\x01\n\x11\x43\x61ncelTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobResponse\"@\x82\xd3\xe4\x93\x02:\"5/v0/physics_ai/training/jobs/{training_job_id}/cancel:\x01*\x12\xd7\x01\n\x0eGetTrainingJob\x12\x45.luminary.proto.api.v0.luminarycloud.physics_ai.GetTrainingJobRequest\x1a\x46.luminary.proto.api.v0.luminarycloud.physics_ai.GetTrainingJobResponse\"6\x82\xd3\xe4\x93\x02\x30\x12./v0/physics_ai/training/jobs/{training_job_id}\x12\xc0\x01\n\rCreateDataset\x12\x44.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetRequest\x1a\x45.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetResponse\"\"\x82\xd3\xe4\x93\x02\x1c\"\x17/v0/physics_ai/datasets:\x01*\x12\xba\x01\n\x0cListDatasets\x12\x43.luminary.proto.api.v0.luminarycloud.physics_ai.ListDatasetsRequest\x1a\x44.luminary.proto.api.v0.luminarycloud.physics_ai.ListDatasetsResponse\"\x1f\x82\xd3\xe4\x93\x02\x19\x12\x17/v0/physics_ai/datasetsB>Z<luminarycloud.com/core/proto/api/v0/luminarycloud/physics_aib\x06proto3')
 
 _PHYSICSAILIFECYCLESTATE = DESCRIPTOR.enum_types_by_name['PhysicsAiLifecycleState']
 PhysicsAiLifecycleState = enum_type_wrapper.EnumTypeWrapper(_PHYSICSAILIFECYCLESTATE)
@@ -43,6 +43,8 @@ MODEL_INITIALIZATION_TYPE_MODEL_VERSION = 2
 MODEL_INITIALIZATION_TYPE_CHECKPOINT = 3
 
 
+_PARAMETERDEFINITION = DESCRIPTOR.message_types_by_name['ParameterDefinition']
+_PARAMETERSCHEMA = DESCRIPTOR.message_types_by_name['ParameterSchema']
 _PHYSICSAIARCHITECTUREVERSION = DESCRIPTOR.message_types_by_name['PhysicsAiArchitectureVersion']
 _PHYSICSAIARCHITECTURE = DESCRIPTOR.message_types_by_name['PhysicsAiArchitecture']
 _LISTARCHITECTURESREQUEST = DESCRIPTOR.message_types_by_name['ListArchitecturesRequest']
@@ -57,10 +59,13 @@ _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY = _GETSOLUTIONDATAPHYSICSA
 _GETSOLUTIONDATAPHYSICSAIRESPONSE = DESCRIPTOR.message_types_by_name['GetSolutionDataPhysicsAIResponse']
 _TRAININGSOLUTION = DESCRIPTOR.message_types_by_name['TrainingSolution']
 _PHYSICSAITRAININGJOB = DESCRIPTOR.message_types_by_name['PhysicsAiTrainingJob']
+_TRAININGINPUTPARAMETER = DESCRIPTOR.message_types_by_name['TrainingInputParameter']
 _SUBMITTRAININGJOBREQUEST = DESCRIPTOR.message_types_by_name['SubmitTrainingJobRequest']
 _SUBMITTRAININGJOBRESPONSE = DESCRIPTOR.message_types_by_name['SubmitTrainingJobResponse']
 _CANCELTRAININGJOBREQUEST = DESCRIPTOR.message_types_by_name['CancelTrainingJobRequest']
 _CANCELTRAININGJOBRESPONSE = DESCRIPTOR.message_types_by_name['CancelTrainingJobResponse']
+_GETTRAININGJOBREQUEST = DESCRIPTOR.message_types_by_name['GetTrainingJobRequest']
+_GETTRAININGJOBRESPONSE = DESCRIPTOR.message_types_by_name['GetTrainingJobResponse']
 _PHYSICSAIDATASET = DESCRIPTOR.message_types_by_name['PhysicsAiDataset']
 _PHYSICSAIDATASETCASEFILE = DESCRIPTOR.message_types_by_name['PhysicsAiDatasetCaseFile']
 _PHYSICSAIDATASETCASEFILES = DESCRIPTOR.message_types_by_name['PhysicsAiDatasetCaseFiles']
@@ -70,6 +75,22 @@ _CREATEDATASETREQUEST = DESCRIPTOR.message_types_by_name['CreateDatasetRequest']
 _CREATEDATASETRESPONSE = DESCRIPTOR.message_types_by_name['CreateDatasetResponse']
 _LISTDATASETSREQUEST = DESCRIPTOR.message_types_by_name['ListDatasetsRequest']
 _LISTDATASETSRESPONSE = DESCRIPTOR.message_types_by_name['ListDatasetsResponse']
+_LISTTRAININGJOBSREQUEST = DESCRIPTOR.message_types_by_name['ListTrainingJobsRequest']
+_LISTTRAININGJOBSRESPONSE = DESCRIPTOR.message_types_by_name['ListTrainingJobsResponse']
+ParameterDefinition = _reflection.GeneratedProtocolMessageType('ParameterDefinition', (_message.Message,), {
+  'DESCRIPTOR' : _PARAMETERDEFINITION,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.ParameterDefinition)
+  })
+_sym_db.RegisterMessage(ParameterDefinition)
+
+ParameterSchema = _reflection.GeneratedProtocolMessageType('ParameterSchema', (_message.Message,), {
+  'DESCRIPTOR' : _PARAMETERSCHEMA,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.ParameterSchema)
+  })
+_sym_db.RegisterMessage(ParameterSchema)
+
 PhysicsAiArchitectureVersion = _reflection.GeneratedProtocolMessageType('PhysicsAiArchitectureVersion', (_message.Message,), {
   'DESCRIPTOR' : _PHYSICSAIARCHITECTUREVERSION,
   '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
@@ -169,6 +190,13 @@ PhysicsAiTrainingJob = _reflection.GeneratedProtocolMessageType('PhysicsAiTraini
   })
 _sym_db.RegisterMessage(PhysicsAiTrainingJob)
 
+TrainingInputParameter = _reflection.GeneratedProtocolMessageType('TrainingInputParameter', (_message.Message,), {
+  'DESCRIPTOR' : _TRAININGINPUTPARAMETER,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.TrainingInputParameter)
+  })
+_sym_db.RegisterMessage(TrainingInputParameter)
+
 SubmitTrainingJobRequest = _reflection.GeneratedProtocolMessageType('SubmitTrainingJobRequest', (_message.Message,), {
   'DESCRIPTOR' : _SUBMITTRAININGJOBREQUEST,
   '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
@@ -197,6 +225,20 @@ CancelTrainingJobResponse = _reflection.GeneratedProtocolMessageType('CancelTrai
   })
 _sym_db.RegisterMessage(CancelTrainingJobResponse)
 
+GetTrainingJobRequest = _reflection.GeneratedProtocolMessageType('GetTrainingJobRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETTRAININGJOBREQUEST,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.GetTrainingJobRequest)
+  })
+_sym_db.RegisterMessage(GetTrainingJobRequest)
+
+GetTrainingJobResponse = _reflection.GeneratedProtocolMessageType('GetTrainingJobResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETTRAININGJOBRESPONSE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.GetTrainingJobResponse)
+  })
+_sym_db.RegisterMessage(GetTrainingJobResponse)
+
 PhysicsAiDataset = _reflection.GeneratedProtocolMessageType('PhysicsAiDataset', (_message.Message,), {
   'DESCRIPTOR' : _PHYSICSAIDATASET,
   '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
@@ -260,6 +302,20 @@ ListDatasetsResponse = _reflection.GeneratedProtocolMessageType('ListDatasetsRes
   })
 _sym_db.RegisterMessage(ListDatasetsResponse)
 
+ListTrainingJobsRequest = _reflection.GeneratedProtocolMessageType('ListTrainingJobsRequest', (_message.Message,), {
+  'DESCRIPTOR' : _LISTTRAININGJOBSREQUEST,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.ListTrainingJobsRequest)
+  })
+_sym_db.RegisterMessage(ListTrainingJobsRequest)
+
+ListTrainingJobsResponse = _reflection.GeneratedProtocolMessageType('ListTrainingJobsResponse', (_message.Message,), {
+  'DESCRIPTOR' : _LISTTRAININGJOBSRESPONSE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.ListTrainingJobsResponse)
+  })
+_sym_db.RegisterMessage(ListTrainingJobsResponse)
+
 _PHYSICSAISERVICE = DESCRIPTOR.services_by_name['PhysicsAiService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -275,70 +331,88 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PHYSICSAISERVICE.methods_by_name['GetSolutionDataPhysicsAI']._serialized_options = b'\202\323\344\223\0020\"+/v0/physics_ai/solutions/{solution_id}/data:\001*'
   _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*'
+  _PHYSICSAISERVICE.methods_by_name['CancelTrainingJob']._options = None
+  _PHYSICSAISERVICE.methods_by_name['CancelTrainingJob']._serialized_options = b'\202\323\344\223\002:\"5/v0/physics_ai/training/jobs/{training_job_id}/cancel:\001*'
+  _PHYSICSAISERVICE.methods_by_name['GetTrainingJob']._options = None
+  _PHYSICSAISERVICE.methods_by_name['GetTrainingJob']._serialized_options = b'\202\323\344\223\0020\022./v0/physics_ai/training/jobs/{training_job_id}'
   _PHYSICSAISERVICE.methods_by_name['CreateDataset']._options = None
   _PHYSICSAISERVICE.methods_by_name['CreateDataset']._serialized_options = b'\202\323\344\223\002\034\"\027/v0/physics_ai/datasets:\001*'
   _PHYSICSAISERVICE.methods_by_name['ListDatasets']._options = None
   _PHYSICSAISERVICE.methods_by_name['ListDatasets']._serialized_options = b'\202\323\344\223\002\031\022\027/v0/physics_ai/datasets'
-  _PHYSICSAILIFECYCLESTATE._serialized_start=5154
-  _PHYSICSAILIFECYCLESTATE._serialized_end=5334
-  _TRAININGDATASOURCETYPE._serialized_start=5337
-  _TRAININGDATASOURCETYPE._serialized_end=5504
-  _MODELINITIALIZATIONTYPE._serialized_start=5507
-  _MODELINITIALIZATIONTYPE._serialized_end=5700
-  _PHYSICSAIARCHITECTUREVERSION._serialized_start=302
-  _PHYSICSAIARCHITECTUREVERSION._serialized_end=475
-  _PHYSICSAIARCHITECTURE._serialized_start=478
-  _PHYSICSAIARCHITECTURE._serialized_end=644
-  _LISTARCHITECTURESREQUEST._serialized_start=646
-  _LISTARCHITECTURESREQUEST._serialized_end=672
-  _LISTARCHITECTURESRESPONSE._serialized_start=674
-  _LISTARCHITECTURESRESPONSE._serialized_end=795
-  _PHYSICSAIMODELVERSION._serialized_start=798
-  _PHYSICSAIMODELVERSION._serialized_end=945
-  _PHYSICSAIMODEL._serialized_start=948
-  _PHYSICSAIMODEL._serialized_end=1100
-  _LISTPRETRAINEDMODELSREQUEST._serialized_start=1102
-  _LISTPRETRAINEDMODELSREQUEST._serialized_end=1131
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1133
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1243
-  _SURFACEGROUP._serialized_start=1245
-  _SURFACEGROUP._serialized_end=1291
-  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_start=1294
-  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_end=1911
-  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_start=1857
-  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_end=1911
-  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_start=1914
-  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_end=2144
-  _TRAININGSOLUTION._serialized_start=2146
-  _TRAININGSOLUTION._serialized_end=2205
-  _PHYSICSAITRAININGJOB._serialized_start=2208
-  _PHYSICSAITRAININGJOB._serialized_end=2894
-  _SUBMITTRAININGJOBREQUEST._serialized_start=2897
-  _SUBMITTRAININGJOBREQUEST._serialized_end=3288
-  _SUBMITTRAININGJOBRESPONSE._serialized_start=3290
-  _SUBMITTRAININGJOBRESPONSE._serialized_end=3409
-  _CANCELTRAININGJOBREQUEST._serialized_start=3411
-  _CANCELTRAININGJOBREQUEST._serialized_end=3462
-  _CANCELTRAININGJOBRESPONSE._serialized_start=3464
-  _CANCELTRAININGJOBRESPONSE._serialized_end=3491
-  _PHYSICSAIDATASET._serialized_start=3494
-  _PHYSICSAIDATASET._serialized_end=3896
-  _PHYSICSAIDATASETCASEFILE._serialized_start=3898
-  _PHYSICSAIDATASETCASEFILE._serialized_end=3957
-  _PHYSICSAIDATASETCASEFILES._serialized_start=3959
-  _PHYSICSAIDATASETCASEFILES._serialized_end=4075
-  _PHYSICSAIDATASETCASE._serialized_start=4078
-  _PHYSICSAIDATASETCASE._serialized_end=4341
-  _CREATEPHYSICSAIDATASETCASEINPUT._serialized_start=4344
-  _CREATEPHYSICSAIDATASETCASEINPUT._serialized_end=4601
-  _CREATEDATASETREQUEST._serialized_start=4604
-  _CREATEDATASETREQUEST._serialized_end=4912
-  _CREATEDATASETRESPONSE._serialized_start=4914
-  _CREATEDATASETRESPONSE._serialized_end=5020
-  _LISTDATASETSREQUEST._serialized_start=5022
-  _LISTDATASETSREQUEST._serialized_end=5043
-  _LISTDATASETSRESPONSE._serialized_start=5045
-  _LISTDATASETSRESPONSE._serialized_end=5151
-  _PHYSICSAISERVICE._serialized_start=5703
-  _PHYSICSAISERVICE._serialized_end=6996
+  _PHYSICSAILIFECYCLESTATE._serialized_start=6020
+  _PHYSICSAILIFECYCLESTATE._serialized_end=6200
+  _TRAININGDATASOURCETYPE._serialized_start=6203
+  _TRAININGDATASOURCETYPE._serialized_end=6370
+  _MODELINITIALIZATIONTYPE._serialized_start=6373
+  _MODELINITIALIZATIONTYPE._serialized_end=6566
+  _PARAMETERDEFINITION._serialized_start=301
+  _PARAMETERDEFINITION._serialized_end=417
+  _PARAMETERSCHEMA._serialized_start=419
+  _PARAMETERSCHEMA._serialized_end=525
+  _PHYSICSAIARCHITECTUREVERSION._serialized_start=528
+  _PHYSICSAIARCHITECTUREVERSION._serialized_end=701
+  _PHYSICSAIARCHITECTURE._serialized_start=704
+  _PHYSICSAIARCHITECTURE._serialized_end=870
+  _LISTARCHITECTURESREQUEST._serialized_start=872
+  _LISTARCHITECTURESREQUEST._serialized_end=898
+  _LISTARCHITECTURESRESPONSE._serialized_start=900
+  _LISTARCHITECTURESRESPONSE._serialized_end=1021
+  _PHYSICSAIMODELVERSION._serialized_start=1024
+  _PHYSICSAIMODELVERSION._serialized_end=1171
+  _PHYSICSAIMODEL._serialized_start=1174
+  _PHYSICSAIMODEL._serialized_end=1326
+  _LISTPRETRAINEDMODELSREQUEST._serialized_start=1328
+  _LISTPRETRAINEDMODELSREQUEST._serialized_end=1357
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1359
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1469
+  _SURFACEGROUP._serialized_start=1471
+  _SURFACEGROUP._serialized_end=1517
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_start=1520
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_end=2137
+  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_start=2083
+  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_end=2137
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_start=2140
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_end=2370
+  _TRAININGSOLUTION._serialized_start=2372
+  _TRAININGSOLUTION._serialized_end=2431
+  _PHYSICSAITRAININGJOB._serialized_start=2434
+  _PHYSICSAITRAININGJOB._serialized_end=3120
+  _TRAININGINPUTPARAMETER._serialized_start=3122
+  _TRAININGINPUTPARAMETER._serialized_end=3179
+  _SUBMITTRAININGJOBREQUEST._serialized_start=3182
+  _SUBMITTRAININGJOBREQUEST._serialized_end=3798
+  _SUBMITTRAININGJOBRESPONSE._serialized_start=3800
+  _SUBMITTRAININGJOBRESPONSE._serialized_end=3919
+  _CANCELTRAININGJOBREQUEST._serialized_start=3921
+  _CANCELTRAININGJOBREQUEST._serialized_end=3972
+  _CANCELTRAININGJOBRESPONSE._serialized_start=3974
+  _CANCELTRAININGJOBRESPONSE._serialized_end=4001
+  _GETTRAININGJOBREQUEST._serialized_start=4003
+  _GETTRAININGJOBREQUEST._serialized_end=4051
+  _GETTRAININGJOBRESPONSE._serialized_start=4053
+  _GETTRAININGJOBRESPONSE._serialized_end=4169
+  _PHYSICSAIDATASET._serialized_start=4172
+  _PHYSICSAIDATASET._serialized_end=4574
+  _PHYSICSAIDATASETCASEFILE._serialized_start=4576
+  _PHYSICSAIDATASETCASEFILE._serialized_end=4635
+  _PHYSICSAIDATASETCASEFILES._serialized_start=4637
+  _PHYSICSAIDATASETCASEFILES._serialized_end=4753
+  _PHYSICSAIDATASETCASE._serialized_start=4756
+  _PHYSICSAIDATASETCASE._serialized_end=5019
+  _CREATEPHYSICSAIDATASETCASEINPUT._serialized_start=5022
+  _CREATEPHYSICSAIDATASETCASEINPUT._serialized_end=5279
+  _CREATEDATASETREQUEST._serialized_start=5282
+  _CREATEDATASETREQUEST._serialized_end=5630
+  _CREATEDATASETRESPONSE._serialized_start=5632
+  _CREATEDATASETRESPONSE._serialized_end=5738
+  _LISTDATASETSREQUEST._serialized_start=5740
+  _LISTDATASETSREQUEST._serialized_end=5761
+  _LISTDATASETSRESPONSE._serialized_start=5763
+  _LISTDATASETSRESPONSE._serialized_end=5869
+  _LISTTRAININGJOBSREQUEST._serialized_start=5871
+  _LISTTRAININGJOBSREQUEST._serialized_end=5896
+  _LISTTRAININGJOBSRESPONSE._serialized_start=5898
+  _LISTTRAININGJOBSRESPONSE._serialized_end=6017
+  _PHYSICSAISERVICE._serialized_start=6569
+  _PHYSICSAISERVICE._serialized_end=8317
 # @@protoc_insertion_point(module_scope)

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

@@ -107,6 +107,56 @@ MODEL_INITIALIZATION_TYPE_CHECKPOINT: ModelInitializationType.ValueType  # 3
 """Initialize from checkpoint."""
 global___ModelInitializationType = ModelInitializationType
 
+class ParameterDefinition(google.protobuf.message.Message):
+    """Structured definition of a parameter in a dataset's parameter schema."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    TYPE_FIELD_NUMBER: builtins.int
+    DIMENSION_FIELD_NUMBER: builtins.int
+    DEFAULT_REFERENCE_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    type: builtins.str
+    """"scalar" or "vector" """
+    dimension: builtins.int
+    """1 for scalar, N for N-dimensional vector"""
+    @property
+    def default_reference(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]:
+        """optional default normalization reference"""
+    description: builtins.str
+    """optional description"""
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        type: builtins.str = ...,
+        dimension: builtins.int = ...,
+        default_reference: collections.abc.Iterable[builtins.float] | None = ...,
+        description: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["default_reference", b"default_reference", "description", b"description", "dimension", b"dimension", "name", b"name", "type", b"type"]) -> None: ...
+
+global___ParameterDefinition = ParameterDefinition
+
+class ParameterSchema(google.protobuf.message.Message):
+    """Container for parameter definitions."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PARAMETERS_FIELD_NUMBER: builtins.int
+    @property
+    def parameters(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ParameterDefinition]: ...
+    def __init__(
+        self,
+        *,
+        parameters: collections.abc.Iterable[global___ParameterDefinition] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["parameters", b"parameters"]) -> None: ...
+
+global___ParameterSchema = ParameterSchema
+
 class PhysicsAiArchitectureVersion(google.protobuf.message.Message):
     """A physics ai architecture version."""
 
@@ -537,6 +587,30 @@ class PhysicsAiTrainingJob(google.protobuf.message.Message):
 
 global___PhysicsAiTrainingJob = PhysicsAiTrainingJob
 
+class TrainingInputParameter(google.protobuf.message.Message):
+    """Input parameter specification for training with reference values for normalization.
+    Type is inherited from the dataset's ParameterSchema.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    REFERENCE_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Parameter name (must exist in dataset's ParameterSchema)"""
+    @property
+    def reference(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]:
+        """Reference value(s) for normalization (optional, falls back to schema default)"""
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        reference: collections.abc.Iterable[builtins.float] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "reference", b"reference"]) -> None: ...
+
+global___TrainingInputParameter = TrainingInputParameter
+
 class SubmitTrainingJobRequest(google.protobuf.message.Message):
     """Request message for submitting a training job"""
 
@@ -550,6 +624,9 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
     INITIALIZATION_TYPE_FIELD_NUMBER: builtins.int
     BASE_MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
     DATASET_ID_FIELD_NUMBER: builtins.int
+    SURFACE_OUTPUTS_FIELD_NUMBER: builtins.int
+    VOLUME_OUTPUTS_FIELD_NUMBER: builtins.int
+    INPUT_PARAMETERS_FIELD_NUMBER: builtins.int
     architecture_version_id: builtins.str
     """Architecture version ID to train."""
     training_description: builtins.str
@@ -560,13 +637,29 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
     def training_solutions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrainingSolution]:
         """Training solutions"""
     training_config: builtins.str
-    """Training configuration in JSON format."""
+    """Training configuration in JSON format. Used for internal/advanced config
+    (resources, priority_class, mode, custom_args). For model I/O configuration,
+    prefer the typed fields below.
+    """
     initialization_type: global___ModelInitializationType.ValueType
     """Model initialization type."""
     base_model_version_id: builtins.str
     """Base model version ID (if initializing from existing model)."""
     dataset_id: builtins.str
     """Dataset ID to use for training (if using dataset-based training)."""
+    @property
+    def surface_outputs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType]:
+        """Model output configuration (typed fields, preferred over JSON training_config).
+        Surface quantities to predict (e.g., PRESSURE_TIME_AVERAGE, WALL_SHEAR_STRESS_TIME_AVERAGE).
+        """
+    @property
+    def volume_outputs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType]:
+        """Volume quantities to predict (e.g., VELOCITY_TIME_AVERAGE, PRESSURE_TIME_AVERAGE)."""
+    @property
+    def input_parameters(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrainingInputParameter]:
+        """Input parameters for the model with reference values for normalization.
+        Parameter names must exist in the dataset's ParameterSchema.
+        """
     def __init__(
         self,
         *,
@@ -578,8 +671,11 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
         initialization_type: global___ModelInitializationType.ValueType = ...,
         base_model_version_id: builtins.str = ...,
         dataset_id: builtins.str = ...,
+        surface_outputs: collections.abc.Iterable[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType] | None = ...,
+        volume_outputs: collections.abc.Iterable[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType] | None = ...,
+        input_parameters: collections.abc.Iterable[global___TrainingInputParameter] | None = ...,
     ) -> 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", "dataset_id", b"dataset_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: ...
+    def ClearField(self, field_name: typing_extensions.Literal["architecture_version_id", b"architecture_version_id", "base_model_version_id", b"base_model_version_id", "dataset_id", b"dataset_id", "external_dataset_uri", b"external_dataset_uri", "initialization_type", b"initialization_type", "input_parameters", b"input_parameters", "surface_outputs", b"surface_outputs", "training_config", b"training_config", "training_description", b"training_description", "training_solutions", b"training_solutions", "volume_outputs", b"volume_outputs"]) -> None: ...
 
 global___SubmitTrainingJobRequest = SubmitTrainingJobRequest
 
@@ -630,6 +726,42 @@ class CancelTrainingJobResponse(google.protobuf.message.Message):
 
 global___CancelTrainingJobResponse = CancelTrainingJobResponse
 
+class GetTrainingJobRequest(google.protobuf.message.Message):
+    """Request message for getting a training job by ID."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOB_ID_FIELD_NUMBER: builtins.int
+    training_job_id: builtins.str
+    """Training job ID to retrieve."""
+    def __init__(
+        self,
+        *,
+        training_job_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["training_job_id", b"training_job_id"]) -> None: ...
+
+global___GetTrainingJobRequest = GetTrainingJobRequest
+
+class GetTrainingJobResponse(google.protobuf.message.Message):
+    """Response message for getting a training job."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOB_FIELD_NUMBER: builtins.int
+    @property
+    def training_job(self) -> global___PhysicsAiTrainingJob:
+        """The 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___GetTrainingJobResponse = GetTrainingJobResponse
+
 class PhysicsAiDataset(google.protobuf.message.Message):
     """A Physics AI dataset"""
 
@@ -824,11 +956,11 @@ class CreateDatasetRequest(google.protobuf.message.Message):
     def cases(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CreatePhysicsAiDatasetCaseInput]:
         """Cases to add to the dataset (case_id will be server-generated)."""
     @property
-    def parameter_schema(self) -> google.protobuf.struct_pb2.Struct:
-        """JSON schema defining allowed parameters for cases in this dataset."""
+    def parameter_schema(self) -> global___ParameterSchema:
+        """Structured parameter schema defining allowed parameters for cases in this dataset."""
     @property
     def export_config(self) -> global___GetSolutionDataPhysicsAIRequest:
-        """Export configuration for extracting solution data forthe cases. The
+        """Export configuration for extracting solution data for the cases. The
         solution_id field is ignored.
         """
     def __init__(
@@ -837,7 +969,7 @@ class CreateDatasetRequest(google.protobuf.message.Message):
         name: builtins.str = ...,
         description: builtins.str = ...,
         cases: collections.abc.Iterable[global___CreatePhysicsAiDatasetCaseInput] | None = ...,
-        parameter_schema: google.protobuf.struct_pb2.Struct | None = ...,
+        parameter_schema: global___ParameterSchema | None = ...,
         export_config: global___GetSolutionDataPhysicsAIRequest | None = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["export_config", b"export_config", "parameter_schema", b"parameter_schema"]) -> builtins.bool: ...
@@ -891,3 +1023,31 @@ class ListDatasetsResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["datasets", b"datasets"]) -> None: ...
 
 global___ListDatasetsResponse = ListDatasetsResponse
+
+class ListTrainingJobsRequest(google.protobuf.message.Message):
+    """Request message for listing training jobs for the current user"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ListTrainingJobsRequest = ListTrainingJobsRequest
+
+class ListTrainingJobsResponse(google.protobuf.message.Message):
+    """Response message for listing training jobs"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOBS_FIELD_NUMBER: builtins.int
+    @property
+    def training_jobs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PhysicsAiTrainingJob]: ...
+    def __init__(
+        self,
+        *,
+        training_jobs: collections.abc.Iterable[global___PhysicsAiTrainingJob] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["training_jobs", b"training_jobs"]) -> None: ...
+
+global___ListTrainingJobsResponse = ListTrainingJobsResponse

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

@@ -35,6 +35,16 @@ class PhysicsAiServiceStub(object):
                 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,
                 )
+        self.CancelTrainingJob = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/CancelTrainingJob',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobResponse.FromString,
+                )
+        self.GetTrainingJob = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/GetTrainingJob',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobResponse.FromString,
+                )
         self.CreateDataset = channel.unary_unary(
                 '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/CreateDataset',
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetRequest.SerializeToString,
@@ -79,6 +89,20 @@ class PhysicsAiServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CancelTrainingJob(self, request, context):
+        """Cancels a Physics AI training job
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetTrainingJob(self, request, context):
+        """Gets a Physics AI training job by ID
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def CreateDataset(self, request, context):
         """Creates a new Physics AI dataset
         """
@@ -116,6 +140,16 @@ def add_PhysicsAiServiceServicer_to_server(servicer, server):
                     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,
             ),
+            'CancelTrainingJob': grpc.unary_unary_rpc_method_handler(
+                    servicer.CancelTrainingJob,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobResponse.SerializeToString,
+            ),
+            'GetTrainingJob': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetTrainingJob,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobResponse.SerializeToString,
+            ),
             'CreateDataset': grpc.unary_unary_rpc_method_handler(
                     servicer.CreateDataset,
                     request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetRequest.FromString,
@@ -205,6 +239,40 @@ class PhysicsAiService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def CancelTrainingJob(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/CancelTrainingJob',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetTrainingJob(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/GetTrainingJob',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def CreateDataset(request,
             target,