luminarycloud 0.22.1__py3-none-any.whl → 0.22.2__py3-none-any.whl

luminarycloud/_proto/client/simulation_pb2.pyi

@@ -9,6 +9,7 @@ import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
 import luminarycloud._proto.base.base_pb2
+import luminarycloud._proto.cad.shape_pb2
 import luminarycloud._proto.client.entity_pb2
 import luminarycloud._proto.entitygroup.entitygroup_pb2
 import luminarycloud._proto.output.output_pb2
@@ -740,6 +741,14 @@ class _PtTableFluidEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._E
     """R410A."""
     REAL_GAS_R507A: _PtTableFluid.ValueType  # 56561
     """R507A."""
+    REAL_GAS_R513A: _PtTableFluid.ValueType  # 13621
+    """R513A."""
+    REAL_GAS_R513B: _PtTableFluid.ValueType  # 5656
+    """R513B."""
+    REAL_GAS_R515A: _PtTableFluid.ValueType  # 55001
+    """R515A."""
+    REAL_GAS_R515B: _PtTableFluid.ValueType  # 5076
+    """R515B."""
     REAL_GAS_RC318: _PtTableFluid.ValueType  # 66529
     """RC318."""
     REAL_GAS_SES36: _PtTableFluid.ValueType  # 14408
@@ -990,6 +999,14 @@ REAL_GAS_R410A: PtTableFluid.ValueType  # 59981
 """R410A."""
 REAL_GAS_R507A: PtTableFluid.ValueType  # 56561
 """R507A."""
+REAL_GAS_R513A: PtTableFluid.ValueType  # 13621
+"""R513A."""
+REAL_GAS_R513B: PtTableFluid.ValueType  # 5656
+"""R513B."""
+REAL_GAS_R515A: PtTableFluid.ValueType  # 55001
+"""R515A."""
+REAL_GAS_R515B: PtTableFluid.ValueType  # 5076
+"""R515B."""
 REAL_GAS_RC318: PtTableFluid.ValueType  # 66529
 """RC318."""
 REAL_GAS_SES36: PtTableFluid.ValueType  # 14408
@@ -4469,7 +4486,7 @@ class ReferenceValues(google.protobuf.message.Message):
 global___ReferenceValues = ReferenceValues
 
 class BoundaryLayerProfile(google.protobuf.message.Message):
-    """TODO: no help"""
+    """Boundary layer meshing parameters to apply to adapted meshes."""
 
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -4502,6 +4519,67 @@ class BoundaryLayerProfile(google.protobuf.message.Message):
 
 global___BoundaryLayerProfile = BoundaryLayerProfile
 
+class ActiveRegion(google.protobuf.message.Message):
+    """Region(s) within which the mesh is adapted at full resolution. Outside
+    of these regions the mesh is coarsened with increasing distance from the
+    region.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ERROR_WEIGHT_FIELD_NUMBER: builtins.int
+    MAX_DISTANCE_FIELD_NUMBER: builtins.int
+    CUBE_FIELD_NUMBER: builtins.int
+    ORIENTED_CUBE_FIELD_NUMBER: builtins.int
+    CYLINDER_FIELD_NUMBER: builtins.int
+    ANNULAR_CYLINDER_FIELD_NUMBER: builtins.int
+    SPHERE_FIELD_NUMBER: builtins.int
+    SPHERE_SHELL_FIELD_NUMBER: builtins.int
+    @property
+    def error_weight(self) -> luminarycloud._proto.base.base_pb2.AdFloatType:
+        """Error weighting from the active region at Max Distance. Local error
+        weighting is 1.0 inside the region and decreases to Error Weight at the Max
+        Distance.
+        """
+    @property
+    def max_distance(self) -> luminarycloud._proto.base.base_pb2.AdFloatType:
+        """Distance from the active regions at which to apply the full error factor."""
+    @property
+    def cube(self) -> luminarycloud._proto.cad.shape_pb2.Cube:
+        """Coordinate-aligned box."""
+    @property
+    def oriented_cube(self) -> luminarycloud._proto.cad.shape_pb2.OrientedCube:
+        """Box oriented along arbitrary axes."""
+    @property
+    def cylinder(self) -> luminarycloud._proto.cad.shape_pb2.Cylinder:
+        """Cylinder shape."""
+    @property
+    def annular_cylinder(self) -> luminarycloud._proto.cad.shape_pb2.AnnularCylinder:
+        """Cylinder with a hole along the axis."""
+    @property
+    def sphere(self) -> luminarycloud._proto.cad.shape_pb2.Sphere:
+        """Sphere shape."""
+    @property
+    def sphere_shell(self) -> luminarycloud._proto.cad.shape_pb2.SphereShell:
+        """Spherical shell shape."""
+    def __init__(
+        self,
+        *,
+        error_weight: luminarycloud._proto.base.base_pb2.AdFloatType | None = ...,
+        max_distance: luminarycloud._proto.base.base_pb2.AdFloatType | None = ...,
+        cube: luminarycloud._proto.cad.shape_pb2.Cube | None = ...,
+        oriented_cube: luminarycloud._proto.cad.shape_pb2.OrientedCube | None = ...,
+        cylinder: luminarycloud._proto.cad.shape_pb2.Cylinder | None = ...,
+        annular_cylinder: luminarycloud._proto.cad.shape_pb2.AnnularCylinder | None = ...,
+        sphere: luminarycloud._proto.cad.shape_pb2.Sphere | None = ...,
+        sphere_shell: luminarycloud._proto.cad.shape_pb2.SphereShell | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["annular_cylinder", b"annular_cylinder", "cube", b"cube", "cylinder", b"cylinder", "error_weight", b"error_weight", "max_distance", b"max_distance", "oriented_cube", b"oriented_cube", "shape", b"shape", "sphere", b"sphere", "sphere_shell", b"sphere_shell"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["annular_cylinder", b"annular_cylinder", "cube", b"cube", "cylinder", b"cylinder", "error_weight", b"error_weight", "max_distance", b"max_distance", "oriented_cube", b"oriented_cube", "shape", b"shape", "sphere", b"sphere", "sphere_shell", b"sphere_shell"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["shape", b"shape"]) -> typing_extensions.Literal["cube", "oriented_cube", "cylinder", "annular_cylinder", "sphere", "sphere_shell"] | None: ...
+
+global___ActiveRegion = ActiveRegion
+
 class AdaptiveMeshRefinement(google.protobuf.message.Message):
     """Adaptive Mesh Refinement"""
 
@@ -4517,6 +4595,7 @@ class AdaptiveMeshRefinement(google.protobuf.message.Message):
     ALL_TET_FIELD_NUMBER: builtins.int
     USER_SCALING_FIELD_NUMBER: builtins.int
     BOUNDARY_LAYER_PROFILE_FIELD_NUMBER: builtins.int
+    ACTIVE_REGION_FIELD_NUMBER: builtins.int
     @property
     def refinement_iterations(self) -> luminarycloud._proto.base.base_pb2.Int:
         """Number of refinement iterations to perform."""
@@ -4546,7 +4625,13 @@ class AdaptiveMeshRefinement(google.protobuf.message.Message):
         """Scale factor between the geometry and the mesh."""
     @property
     def boundary_layer_profile(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BoundaryLayerProfile]:
-        """TODO: no help"""
+        """Boundary layer meshing parameters to apply to adapted meshes."""
+    @property
+    def active_region(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ActiveRegion]:
+        """Region(s) within which the mesh is adapted at full resolution. Outside
+        of these regions the mesh is coarsened with increasing distance from the
+        region.
+        """
     def __init__(
         self,
         *,
@@ -4560,9 +4645,10 @@ class AdaptiveMeshRefinement(google.protobuf.message.Message):
         all_tet: global___AllTet.ValueType = ...,
         user_scaling: luminarycloud._proto.base.base_pb2.AdFloatType | None = ...,
         boundary_layer_profile: collections.abc.Iterable[global___BoundaryLayerProfile] | None = ...,
+        active_region: collections.abc.Iterable[global___ActiveRegion] | None = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["final_target_complexity", b"final_target_complexity", "initial_target_complexity", b"initial_target_complexity", "max_refinement_interval", b"max_refinement_interval", "refinement_dispatch_interval", b"refinement_dispatch_interval", "refinement_iterations", b"refinement_iterations", "target_cv_millions", b"target_cv_millions", "user_scaling", b"user_scaling"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["all_tet", b"all_tet", "boundary_layer_profile", b"boundary_layer_profile", "final_target_complexity", b"final_target_complexity", "initial_target_complexity", b"initial_target_complexity", "max_refinement_interval", b"max_refinement_interval", "meshing_method", b"meshing_method", "refinement_dispatch_interval", b"refinement_dispatch_interval", "refinement_iterations", b"refinement_iterations", "target_cv_millions", b"target_cv_millions", "user_scaling", b"user_scaling"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["active_region", b"active_region", "all_tet", b"all_tet", "boundary_layer_profile", b"boundary_layer_profile", "final_target_complexity", b"final_target_complexity", "initial_target_complexity", b"initial_target_complexity", "max_refinement_interval", b"max_refinement_interval", "meshing_method", b"meshing_method", "refinement_dispatch_interval", b"refinement_dispatch_interval", "refinement_iterations", b"refinement_iterations", "target_cv_millions", b"target_cv_millions", "user_scaling", b"user_scaling"]) -> None: ...
 
 global___AdaptiveMeshRefinement = AdaptiveMeshRefinement
 

luminarycloud/_proto/physicsaiinferenceservice/physicsaiinferenceservice_pb2.py

@@ -12,11 +12,12 @@ from google.protobuf import symbol_database as _symbol_database
 _sym_db = _symbol_database.Default()
 
 
-from luminarycloud._proto.base import base_pb2 as proto_dot_base_dot_base__pb2
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
 from luminarycloud._proto.api.v0.luminarycloud.physicsaiinference import physicsaiinference_pb2 as proto_dot_api_dot_v0_dot_luminarycloud_dot_physicsaiinference_dot_physicsaiinference__pb2
+from luminarycloud._proto.ratelimit import ratelimit_pb2 as proto_dot_ratelimit_dot_ratelimit__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n?proto/physicsaiinferenceservice/physicsaiinferenceservice.proto\x12(luminary.proto.physicsaiinferenceservice\x1a\x15proto/base/base.proto\x1a\x46proto/api/v0/luminarycloud/physicsaiinference/physicsaiinference.proto2\xda\x06\n\x19PhysicsAiInferenceService\x12\xcd\x01\n\x19\x43reateInferenceServiceJob\x12X.luminary.proto.api.v0.luminarycloud.physicsaiinference.CreateInferenceServiceJobRequest\x1aV.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobResponse\x12\xd2\x01\n\x1e\x43reateInferenceServiceJobAsync\x12X.luminary.proto.api.v0.luminarycloud.physicsaiinference.CreateInferenceServiceJobRequest\x1aV.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobResponse\x12\xc7\x01\n\x16GetInferenceServiceJob\x12U.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobRequest\x1aV.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobResponse\x12\xcd\x01\n\x18ListInferenceServiceJobs\x12W.luminary.proto.api.v0.luminarycloud.physicsaiinference.ListInferenceServiceJobsRequest\x1aX.luminary.proto.api.v0.luminarycloud.physicsaiinference.ListInferenceServiceJobsResponseB8Z6luminarycloud.com/core/proto/physicsaiinferenceserviceb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n?proto/physicsaiinferenceservice/physicsaiinferenceservice.proto\x12(luminary.proto.physicsaiinferenceservice\x1a\x1bgoogle/protobuf/empty.proto\x1a\x46proto/api/v0/luminarycloud/physicsaiinference/physicsaiinference.proto\x1a\x1fproto/ratelimit/ratelimit.proto2\xb1\t\n\x19PhysicsAiInferenceService\x12\xd9\x01\n\x19\x43reateInferenceServiceJob\x12X.luminary.proto.api.v0.luminarycloud.physicsaiinference.CreateInferenceServiceJobRequest\x1aV.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobResponse\"\n\x8a\xb5\x18\x06\x08\n\x12\x02\x08\x01\x12\xde\x01\n\x1e\x43reateInferenceServiceJobAsync\x12X.luminary.proto.api.v0.luminarycloud.physicsaiinference.CreateInferenceServiceJobRequest\x1aV.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobResponse\"\n\x8a\xb5\x18\x06\x08\n\x12\x02\x08\x01\x12\xbb\x01\n\x12GetInferenceFields\x12Q.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceFieldsRequest\x1aR.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceFieldsResponse\x12\xc7\x01\n\x16GetInferenceServiceJob\x12U.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobRequest\x1aV.luminary.proto.api.v0.luminarycloud.physicsaiinference.GetInferenceServiceJobResponse\x12\xcd\x01\n\x18ListInferenceServiceJobs\x12W.luminary.proto.api.v0.luminarycloud.physicsaiinference.ListInferenceServiceJobsRequest\x1aX.luminary.proto.api.v0.luminarycloud.physicsaiinference.ListInferenceServiceJobsResponse\x12\x7f\n\x12\x44\x65leteInferenceJob\x12Q.luminary.proto.api.v0.luminarycloud.physicsaiinference.DeleteInferenceJobRequest\x1a\x16.google.protobuf.EmptyB8Z6luminarycloud.com/core/proto/physicsaiinferenceserviceb\x06proto3')
 
 
 
@@ -25,6 +26,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'Z6luminarycloud.com/core/proto/physicsaiinferenceservice'
-  _PHYSICSAIINFERENCESERVICE._serialized_start=205
-  _PHYSICSAIINFERENCESERVICE._serialized_end=1063
+  _PHYSICSAIINFERENCESERVICE.methods_by_name['CreateInferenceServiceJob']._options = None
+  _PHYSICSAIINFERENCESERVICE.methods_by_name['CreateInferenceServiceJob']._serialized_options = b'\212\265\030\006\010\n\022\002\010\001'
+  _PHYSICSAIINFERENCESERVICE.methods_by_name['CreateInferenceServiceJobAsync']._options = None
+  _PHYSICSAIINFERENCESERVICE.methods_by_name['CreateInferenceServiceJobAsync']._serialized_options = b'\212\265\030\006\010\n\022\002\010\001'
+  _PHYSICSAIINFERENCESERVICE._serialized_start=244
+  _PHYSICSAIINFERENCESERVICE._serialized_end=1445
 # @@protoc_insertion_point(module_scope)

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2.py

@@ -13,9 +13,10 @@ _sym_db = _symbol_database.Default()
 
 
 from luminarycloud._proto.api.v0.luminarycloud.physics_ai import physics_ai_pb2 as proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2
+from luminarycloud._proto.ratelimit import ratelimit_pb2 as proto_dot_ratelimit_dot_ratelimit__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n=proto/physicsaitrainingservice/physicsaitrainingservice.proto\x12\'luminary.proto.physicsaitrainingservice\x1a\x36proto/api/v0/luminarycloud/physics_ai/physics_ai.proto2\xf4\x02\n\x18PhysicsAiTrainingService\x12\xaa\x01\n\x11SubmitTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobResponse\"\x00\x12\xaa\x01\n\x11\x43\x61ncelTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobResponse\"\x00\x42\x37Z5luminarycloud.com/core/proto/physicsaitrainingserviceb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n=proto/physicsaitrainingservice/physicsaitrainingservice.proto\x12\'luminary.proto.physicsaitrainingservice\x1a\x36proto/api/v0/luminarycloud/physics_ai/physics_ai.proto\x1a\x1fproto/ratelimit/ratelimit.proto2\x9f\x04\n\x18PhysicsAiTrainingService\x12\x9e\x01\n\rCreateDataset\x12\x44.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetRequest\x1a\x45.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetResponse\"\x00\x12\xb4\x01\n\x11SubmitTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobResponse\"\n\x8a\xb5\x18\x06\x08\x05\x12\x02\x08\x01\x12\xaa\x01\n\x11\x43\x61ncelTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobResponse\"\x00\x42\x37Z5luminarycloud.com/core/proto/physicsaitrainingserviceb\x06proto3')
 
 
 
@@ -24,6 +25,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'Z5luminarycloud.com/core/proto/physicsaitrainingservice'
-  _PHYSICSAITRAININGSERVICE._serialized_start=163
-  _PHYSICSAITRAININGSERVICE._serialized_end=535
+  _PHYSICSAITRAININGSERVICE.methods_by_name['SubmitTrainingJob']._options = None
+  _PHYSICSAITRAININGSERVICE.methods_by_name['SubmitTrainingJob']._serialized_options = b'\212\265\030\006\010\005\022\002\010\001'
+  _PHYSICSAITRAININGSERVICE._serialized_start=196
+  _PHYSICSAITRAININGSERVICE._serialized_end=739
 # @@protoc_insertion_point(module_scope)

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2_grpc.py

@@ -15,6 +15,11 @@ class PhysicsAiTrainingServiceStub(object):
         Args:
             channel: A grpc.Channel.
         """
+        self.CreateDataset = channel.unary_unary(
+                '/luminary.proto.physicsaitrainingservice.PhysicsAiTrainingService/CreateDataset',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetResponse.FromString,
+                )
         self.SubmitTrainingJob = channel.unary_unary(
                 '/luminary.proto.physicsaitrainingservice.PhysicsAiTrainingService/SubmitTrainingJob',
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobRequest.SerializeToString,
@@ -31,6 +36,13 @@ class PhysicsAiTrainingServiceServicer(object):
     """PhysicsAiTrainingService provides training functionality for Physics AI
     """
 
+    def CreateDataset(self, request, context):
+        """Create a Physics AI dataset
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def SubmitTrainingJob(self, request, context):
         """Submit a physics AI training job
         """
@@ -48,6 +60,11 @@ class PhysicsAiTrainingServiceServicer(object):
 
 def add_PhysicsAiTrainingServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
+            '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,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetResponse.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,
@@ -69,6 +86,23 @@ class PhysicsAiTrainingService(object):
     """PhysicsAiTrainingService provides training functionality for Physics AI
     """
 
+    @staticmethod
+    def CreateDataset(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.physicsaitrainingservice.PhysicsAiTrainingService/CreateDataset',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def SubmitTrainingJob(request,
             target,

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2_grpc.pyi

@@ -10,6 +10,11 @@ class PhysicsAiTrainingServiceStub:
     """PhysicsAiTrainingService provides training functionality for Physics AI"""
 
     def __init__(self, channel: grpc.Channel) -> None: ...
+    CreateDataset: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetRequest,
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetResponse,
+    ]
+    """Create a Physics AI dataset"""
     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,
@@ -24,6 +29,13 @@ class PhysicsAiTrainingServiceStub:
 class PhysicsAiTrainingServiceServicer(metaclass=abc.ABCMeta):
     """PhysicsAiTrainingService provides training functionality for Physics AI"""
 
+    @abc.abstractmethod
+    def CreateDataset(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetResponse:
+        """Create a Physics AI dataset"""
     @abc.abstractmethod
     def SubmitTrainingJob(
         self,

luminarycloud/_wrapper.py

@@ -103,20 +103,62 @@ class ProtoWrapper(Generic[P]):
         def getter(field_name: str) -> Any:
             return lambda self: getattr(self._proto, field_name)
 
-        def wrapped_getter(field_name: str, wrapper: type[ProtoWrapperBase]) -> Any:
-            return lambda self: wrapper(getattr(self._proto, field_name))
+        def wrapped_getter(field_name: str, wrapper: type) -> Any:
+            def _get(self):
+                proto_value = getattr(self._proto, field_name)
+
+                # If it's a ProtoWrapperBase, use the standard pattern
+                if issubclass(wrapper, ProtoWrapperBase):
+                    return wrapper(proto_value)
+
+                # If it's Vector3, use the conversion pattern
+                if wrapper is Vector3:
+                    return Vector3(
+                        x=float(proto_value.x), y=float(proto_value.y), z=float(proto_value.z)
+                    )
+
+                # For other types, try the constructor directly
+                try:
+                    return wrapper(proto_value)
+                except (TypeError, ValueError):
+                    # Fallback: return the proto value as-is
+                    return proto_value
+
+            return _get
 
         # This binds the field name to the setter.
         def setter(field_name: str) -> Callable[[C, Any], None]:
             return lambda self, value: setattr(self._proto, field_name, value)
 
-        def wrapped_setter(
-            field_name: str, wrapper: type[ProtoWrapperBase]
-        ) -> Callable[[C, Any], None]:
+        def wrapped_setter(field_name: str, wrapper: type) -> Callable[[C, Any], None]:
             def _set(self: C, value: Any) -> None:
                 if not isinstance(value, wrapper):
                     raise TypeError(f"{field_name} should be a {wrapper.__name__}")
-                setattr(self._proto, field_name, value._proto)
+
+                # If it's a ProtoWrapperBase, extract the _proto
+                if issubclass(wrapper, ProtoWrapperBase):
+                    # For protobuf message fields, use CopyFrom instead of direct assignment
+                    proto_field = getattr(self._proto, field_name)
+                    proto_field.CopyFrom(value._proto)
+
+                # If it's Vector3, convert to protobuf Vector3
+                elif wrapper is Vector3:
+                    proto_field = getattr(self._proto, field_name)
+                    proto_field.x = float(value.x)
+                    proto_field.y = float(value.y)
+                    proto_field.z = float(value.z)
+
+                # For other types, try direct assignment or conversion
+                else:
+                    try:
+                        # Try using _to_proto method if it exists
+                        if hasattr(value, "_to_proto"):
+                            setattr(self._proto, field_name, value._to_proto())
+                        else:
+                            setattr(self._proto, field_name, value)
+                    except (TypeError, ValueError, AttributeError):
+                        # Fallback: direct assignment
+                        setattr(self._proto, field_name, value)
 
             return _set
 
@@ -148,7 +190,11 @@ class ProtoWrapper(Generic[P]):
                     pass
                 else:
                     try:
-                        if issubclass(_type, Enum) or issubclass(_type, ProtoWrapperBase):
+                        if (
+                            issubclass(_type, Enum)
+                            or issubclass(_type, ProtoWrapperBase)
+                            or _type is Vector3
+                        ):
                             fget = wrapped_getter(field.name, _type)
                             fset = wrapped_setter(field.name, _type)
                     except TypeError:

luminarycloud/feature_modification.py

@@ -148,6 +148,29 @@ def _update_repeated_field(
     target_field.extend(new_values)
 
 
+def _update_create_op_from_shape(create_op: gpb.Create, shape: Shape) -> None:
+    """Helper function to replace the shape in a gpb.Create operation with a shape object.
+
+    Args:
+        create: The gpb.Create to modify
+        shape: The shape to add
+    """
+    if isinstance(shape, Sphere):
+        create_op.sphere.CopyFrom(shape._to_proto())
+    elif isinstance(shape, Cube):
+        create_op.box.CopyFrom(shape._to_proto())
+    elif isinstance(shape, Cylinder):
+        create_op.cylinder.CopyFrom(shape._to_proto())
+    elif isinstance(shape, Torus):
+        create_op.torus.CopyFrom(shape._to_proto())
+    elif isinstance(shape, Cone):
+        create_op.cone.CopyFrom(shape._to_proto())
+    elif isinstance(shape, HalfSphere):
+        create_op.half_sphere.CopyFrom(shape._to_proto())
+    else:
+        raise TypeError(f"Unsupported shape type: {type(shape)}")
+
+
 def modify_import(
     feature: gpb.Feature,
     geometry_url: Optional[str] = None,
@@ -205,22 +228,7 @@ def modify_create(
     create_op = feature_copy.create
 
     if shape is not None:
-        create_op.ClearField("shape")
-
-        if isinstance(shape, Sphere):
-            create_op.sphere.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Cube):
-            create_op.box.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Cylinder):
-            create_op.cylinder.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Torus):
-            create_op.torus.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Cone):
-            create_op.cone.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, HalfSphere):
-            create_op.half_sphere.CopyFrom(shape._to_proto())  # type: ignore
-        else:
-            raise TypeError(f"Unsupported shape type: {type(shape)}")
+        _update_create_op_from_shape(create_op, shape)
 
     return gpb.Modification(
         mod_type=gpb.Modification.ModificationType.MODIFICATION_TYPE_UPDATE_FEATURE,
@@ -719,22 +727,7 @@ def modify_farfield(
     farfield_op = feature_copy.farfield
 
     if shape is not None:
-        create_op = gpb.Create()
-        if isinstance(shape, Sphere):
-            create_op.sphere.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Cube):
-            create_op.box.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Cylinder):
-            create_op.cylinder.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Torus):
-            create_op.torus.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, Cone):
-            create_op.cone.CopyFrom(shape._to_proto())  # type: ignore
-        elif isinstance(shape, HalfSphere):
-            create_op.half_sphere.CopyFrom(shape._to_proto())  # type: ignore
-        else:
-            raise TypeError(f"Unsupported shape type: {type(shape)}")
-        farfield_op.create.CopyFrom(create_op)
+        _update_create_op_from_shape(farfield_op.create, shape)
 
     if volumes is not None:
         vol_ids = _volumes_to_int_list(volumes)

luminarycloud/geometry.py

@@ -178,17 +178,17 @@ class Geometry(ProtoWrapperBase):
         """
         create_proto = gpb.Create()
         if isinstance(shape, shapes.Sphere):
-            create_proto.sphere.CopyFrom(shape._to_proto())  # type: ignore
+            create_proto.sphere.CopyFrom(shape._to_proto())
         elif isinstance(shape, shapes.Cube):
-            create_proto.box.CopyFrom(shape._to_proto())  # type: ignore
+            create_proto.box.CopyFrom(shape._to_proto())
         elif isinstance(shape, shapes.Cylinder):
-            create_proto.cylinder.CopyFrom(shape._to_proto())  # type: ignore
+            create_proto.cylinder.CopyFrom(shape._to_proto())
         elif isinstance(shape, shapes.Torus):
-            create_proto.torus.CopyFrom(shape._to_proto())  # type: ignore
+            create_proto.torus.CopyFrom(shape._to_proto())
         elif isinstance(shape, shapes.Cone):
-            create_proto.cone.CopyFrom(shape._to_proto())  # type: ignore
+            create_proto.cone.CopyFrom(shape._to_proto())
         elif isinstance(shape, shapes.HalfSphere):
-            create_proto.half_sphere.CopyFrom(shape._to_proto())  # type: ignore
+            create_proto.half_sphere.CopyFrom(shape._to_proto())
         else:
             raise TypeError(f"Unsupported shape for farfield: {type(shape)}")
         self._modify(

luminarycloud/outputs/__init__.py

@@ -19,6 +19,8 @@ from .output_definitions import (
 
 from .stopping_conditions import (
     StoppingCondition as StoppingCondition,
+    StoppingConditionResult as StoppingConditionResult,
+    StoppingConditionStatusResult as StoppingConditionStatusResult,
     GeneralStoppingConditions as GeneralStoppingConditions,
     create_or_update_stopping_condition as create_or_update_stopping_condition,
     get_stopping_condition as get_stopping_condition,

luminarycloud/outputs/output_definitions.py

@@ -125,7 +125,7 @@ class SurfaceAverageOutputDefinition(CodeRepr):
     space_averaging_type: SpaceAveragingType = SpaceAveragingType.AREA
     "Type of spatial averaging to use."
     vector_component: Vector3Component = Vector3Component.UNSPECIFIED
-    "For vector quantities, the component to measure."
+    "For vector quantities, the component to measure must be specified (X, Y, or Z)."
 
     def _to_proto(self) -> feoutputpb.OutputNode:
         proto = feoutputpb.OutputNode(
@@ -309,7 +309,7 @@ class PointProbeOutputDefinition(CodeRepr):
     include: OutputDefinitionInclusions = field(default_factory=OutputDefinitionInclusions)
     "Configuration for what values to include."
     vector_component: Vector3Component = Vector3Component.UNSPECIFIED
-    "For vector quantities, the component to measure."
+    "For vector quantities, the component to measure must be specified (X, Y, or Z)."
 
     def _to_proto(self) -> feoutputpb.OutputNode:
         proto = feoutputpb.OutputNode(
@@ -356,7 +356,7 @@ class VolumeReductionOutputDefinition(CodeRepr):
     volume_reduction_type: VolumeReductionType = VolumeReductionType.AVERAGE
     "Type of reduction to perform for each volume."
     vector_component: Vector3Component = Vector3Component.UNSPECIFIED
-    "For vector quantities, the component to measure."
+    "For vector quantities, the component to measure must be specified (X, Y, or Z)."
 
     def _to_proto(self) -> feoutputpb.OutputNode:
         proto = feoutputpb.OutputNode(

luminarycloud/outputs/stopping_conditions.py

@@ -2,6 +2,7 @@
 
 from .._client import get_default_client
 from .._proto.api.v0.luminarycloud.stopping_condition import stopping_condition_pb2 as stopcondpb
+from .._proto.api.v0.luminarycloud.simulation import simulation_pb2 as simulationpb
 from .._proto.output import output_pb2 as outputpb
 from dataclasses import dataclass
 
@@ -213,3 +214,96 @@ def update_general_stopping_conditions(
     )
     res = get_default_client().UpdateBasicStoppingConditions(req)
     return GeneralStoppingConditions._from_proto(res.basic_stopping_conditions)
+
+
+@dataclass
+class StoppingConditionResult:
+    """
+    Result of evaluating a single stopping condition.
+
+    Attributes
+    ----------
+    output_name : str
+        Name of the output being monitored.
+    threshold : float
+        The threshold value for this condition.
+    value : float
+        The actual value of the monitored output.
+    satisfied : bool
+        Whether this condition was satisfied.
+    is_force_condition : bool
+        Whether this is a force-stop condition (e.g., max iterations, max time).
+        Force conditions cause immediate termination regardless of other conditions.
+    """
+
+    output_name: str
+    threshold: float
+    value: float
+    satisfied: bool
+    is_force_condition: bool
+
+
+@dataclass
+class StoppingConditionStatusResult:
+    """
+    Status of all stopping conditions for a simulation.
+
+    This represents the evaluation of stopping conditions against the final
+    simulation results.
+
+    Attributes
+    ----------
+    overall_success : bool
+        Whether the overall stopping criteria were met (excluding force-stop conditions).
+    force_stopped : bool
+        Whether a force-stop condition was triggered.
+    condition_results : list[StoppingConditionResult]
+        Results for each individual stopping condition.
+    """
+
+    overall_success: bool
+    force_stopped: bool
+    condition_results: list[StoppingConditionResult]
+
+    @classmethod
+    def _from_proto(
+        cls, proto: simulationpb.GetStoppingConditionStatusResponse
+    ) -> "StoppingConditionStatusResult":
+        condition_results = []
+        for i, condition in enumerate(proto.conditions):
+            # Get the value from the status with bounds checking
+            value = (
+                proto.status.value[i].values[0].value
+                if i < len(proto.status.value) and proto.status.value[i].values
+                else float("nan")
+            )
+
+            # Get the threshold, checking for field presence (not truthiness)
+            threshold = condition.threshold.value if condition.HasField("threshold") else 0.0
+
+            # Check if this is a force condition
+            is_force = condition.op == outputpb.STOP_COND_OP_FORCE
+
+            # Get output name with field presence check
+            output_name = condition.output.name if condition.HasField("output") else ""
+
+            # Get satisfied status with bounds checking
+            satisfied = (
+                proto.status.cond_success[i] if i < len(proto.status.cond_success) else False
+            )
+
+            condition_results.append(
+                StoppingConditionResult(
+                    output_name=output_name,
+                    threshold=threshold,
+                    value=value,
+                    satisfied=satisfied,
+                    is_force_condition=is_force,
+                )
+            )
+
+        return cls(
+            overall_success=proto.status.success,
+            force_stopped=proto.status.force_stop,
+            condition_results=condition_results,
+        )