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

luminarycloud/_proto/client/simulation_pb2.pyi

@@ -599,6 +599,31 @@ TEMPERATURE_DEPENDENT_LAMINAR_VISCOSITY: LaminarViscosityModelNewtonian.ValueTyp
 """Tabulated dynamic viscosity values vs temperature."""
 global___LaminarViscosityModelNewtonian = LaminarViscosityModelNewtonian
 
+class _ThermalConductivityModelSolid:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _ThermalConductivityModelSolidEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ThermalConductivityModelSolid.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID_THERMAL_CONDUCTIVITY_MODEL_SOLID: _ThermalConductivityModelSolid.ValueType  # 0
+    ISOTROPIC_CONDUCTIVITY: _ThermalConductivityModelSolid.ValueType  # 52316
+    """Equal thermal conductivity in all directions."""
+    ORTHOTROPIC_CONDUCTIVITY: _ThermalConductivityModelSolid.ValueType  # 60601
+    """Different thermal conductivity along each axis of the frames to which
+    the volumes of this material are assigned.
+    """
+
+class ThermalConductivityModelSolid(_ThermalConductivityModelSolid, metaclass=_ThermalConductivityModelSolidEnumTypeWrapper): ...
+
+INVALID_THERMAL_CONDUCTIVITY_MODEL_SOLID: ThermalConductivityModelSolid.ValueType  # 0
+ISOTROPIC_CONDUCTIVITY: ThermalConductivityModelSolid.ValueType  # 52316
+"""Equal thermal conductivity in all directions."""
+ORTHOTROPIC_CONDUCTIVITY: ThermalConductivityModelSolid.ValueType  # 60601
+"""Different thermal conductivity along each axis of the frames to which
+the volumes of this material are assigned.
+"""
+global___ThermalConductivityModelSolid = ThermalConductivityModelSolid
+
 class _MaterialSolidPreset:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
@@ -3086,7 +3111,9 @@ class MaterialSolid(google.protobuf.message.Message):
 
     CONSTANT_DENSITY_VALUE_SOLID_FIELD_NUMBER: builtins.int
     SPECIFIC_HEAT_CP_SOLID_FIELD_NUMBER: builtins.int
+    THERMAL_CONDUCTIVITY_MODEL_SOLID_FIELD_NUMBER: builtins.int
     THERMAL_CONDUCTIVITY_CONSTANT_SOLID_FIELD_NUMBER: builtins.int
+    THERMAL_CONDUCTIVITY_ORTHOTROPIC_SOLID_FIELD_NUMBER: builtins.int
     THERMAL_CONDUCTIVITY_TABLE_DATA_FIELD_NUMBER: builtins.int
     MATERIAL_SOLID_PRESET_FIELD_NUMBER: builtins.int
     @property
@@ -3094,10 +3121,15 @@ class MaterialSolid(google.protobuf.message.Message):
         """Constant density value."""
     @property
     def specific_heat_cp_solid(self) -> luminarycloud._proto.base.base_pb2.AdFloatType:
-        """Specific heat at constant pressure."""
+        """Specific heat."""
+    thermal_conductivity_model_solid: global___ThermalConductivityModelSolid.ValueType
+    """Uniform or direction-dependent conductivity models."""
     @property
     def thermal_conductivity_constant_solid(self) -> luminarycloud._proto.base.base_pb2.AdFloatType:
         """The thermal conductivity of the material."""
+    @property
+    def thermal_conductivity_orthotropic_solid(self) -> luminarycloud._proto.base.base_pb2.AdVector3:
+        """The thermal conductivities of the material along each axis."""
     thermal_conductivity_table_data: builtins.str
     """Correlation between thermal conductivity and temperature."""
     material_solid_preset: global___MaterialSolidPreset.ValueType
@@ -3107,12 +3139,14 @@ class MaterialSolid(google.protobuf.message.Message):
         *,
         constant_density_value_solid: luminarycloud._proto.base.base_pb2.AdFloatType | None = ...,
         specific_heat_cp_solid: luminarycloud._proto.base.base_pb2.AdFloatType | None = ...,
+        thermal_conductivity_model_solid: global___ThermalConductivityModelSolid.ValueType = ...,
         thermal_conductivity_constant_solid: luminarycloud._proto.base.base_pb2.AdFloatType | None = ...,
+        thermal_conductivity_orthotropic_solid: luminarycloud._proto.base.base_pb2.AdVector3 | None = ...,
         thermal_conductivity_table_data: builtins.str = ...,
         material_solid_preset: global___MaterialSolidPreset.ValueType = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["constant_density_value_solid", b"constant_density_value_solid", "specific_heat_cp_solid", b"specific_heat_cp_solid", "thermal_conductivity_constant_solid", b"thermal_conductivity_constant_solid"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["constant_density_value_solid", b"constant_density_value_solid", "material_solid_preset", b"material_solid_preset", "specific_heat_cp_solid", b"specific_heat_cp_solid", "thermal_conductivity_constant_solid", b"thermal_conductivity_constant_solid", "thermal_conductivity_table_data", b"thermal_conductivity_table_data"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["constant_density_value_solid", b"constant_density_value_solid", "specific_heat_cp_solid", b"specific_heat_cp_solid", "thermal_conductivity_constant_solid", b"thermal_conductivity_constant_solid", "thermal_conductivity_orthotropic_solid", b"thermal_conductivity_orthotropic_solid"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["constant_density_value_solid", b"constant_density_value_solid", "material_solid_preset", b"material_solid_preset", "specific_heat_cp_solid", b"specific_heat_cp_solid", "thermal_conductivity_constant_solid", b"thermal_conductivity_constant_solid", "thermal_conductivity_model_solid", b"thermal_conductivity_model_solid", "thermal_conductivity_orthotropic_solid", b"thermal_conductivity_orthotropic_solid", "thermal_conductivity_table_data", b"thermal_conductivity_table_data"]) -> None: ...
 
 global___MaterialSolid = MaterialSolid
 

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: proto/physicsaitrainingservice/physicsaitrainingservice.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_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
+
+
+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\xc7\x01\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\x42\x37Z5luminarycloud.com/core/proto/physicsaitrainingserviceb\x06proto3')
+
+
+
+_PHYSICSAITRAININGSERVICE = DESCRIPTOR.services_by_name['PhysicsAiTrainingService']
+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=362
+# @@protoc_insertion_point(module_scope)

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2.pyi

@@ -0,0 +1,7 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2025 Luminary Cloud, Inc. All Rights Reserved."""
+import google.protobuf.descriptor
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2_grpc.py

@@ -0,0 +1,70 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+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
+
+
+class PhysicsAiTrainingServiceStub(object):
+    """PhysicsAiTrainingService provides training functionality for Physics AI
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        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,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobResponse.FromString,
+                )
+
+
+class PhysicsAiTrainingServiceServicer(object):
+    """PhysicsAiTrainingService provides training functionality for Physics AI
+    """
+
+    def SubmitTrainingJob(self, request, context):
+        """Submit a physics AI training job
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_PhysicsAiTrainingServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            '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.physicsaitrainingservice.PhysicsAiTrainingService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class PhysicsAiTrainingService(object):
+    """PhysicsAiTrainingService provides training functionality for Physics AI
+    """
+
+    @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.physicsaitrainingservice.PhysicsAiTrainingService/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/physicsaitrainingservice/physicsaitrainingservice_pb2_grpc.pyi

@@ -0,0 +1,30 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2025 Luminary Cloud, Inc. All Rights Reserved."""
+import abc
+import grpc
+import luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2
+
+class PhysicsAiTrainingServiceStub:
+    """PhysicsAiTrainingService provides training functionality for Physics AI"""
+
+    def __init__(self, channel: grpc.Channel) -> None: ...
+    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,
+    ]
+    """Submit a physics AI training job"""
+
+class PhysicsAiTrainingServiceServicer(metaclass=abc.ABCMeta):
+    """PhysicsAiTrainingService provides training functionality for Physics AI"""
+
+    @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:
+        """Submit a physics AI training job"""
+
+def add_PhysicsAiTrainingServiceServicer_to_server(servicer: PhysicsAiTrainingServiceServicer, server: grpc.Server) -> None: ...

luminarycloud/exceptions.py

@@ -22,6 +22,12 @@ class SDKException(Exception):
         return [self.message]
 
 
+class Timeout(SDKException):
+    """Raised when some long-running operation doesn't finish before its deadline."""
+
+    pass
+
+
 class RpcError(SDKException):
     """Raised when an RPC error occurs."""
 
@@ -70,6 +76,6 @@ class FailedPreconditionError(RpcError):
 
 
 class DeadlineExceededError(RpcError):
-    """Raised when the deadline expired before the operation could complete.  I.e. it timed out."""
+    """Raised when the gRPC deadline expired before the operation could complete.  I.e. it timed out."""
 
     pass

luminarycloud/geometry.py

@@ -42,6 +42,8 @@ class Geometry(ProtoWrapperBase):
     "Geometry name."
     status: GeometryStatus
     "The status of the geometry."
+    project_id: ProjectID
+    "The ID of the project this geometry belongs to."
 
     _proto: geometrypb.Geometry
 
@@ -63,7 +65,7 @@ class Geometry(ProtoWrapperBase):
         """
         Get the project this geometry belongs to.
         """
-        return lc.get_project(ProjectID(self._proto.project_id))
+        return lc.get_project(self.project_id)
 
     def update(self, *, name: str | None = None) -> None:
         """

luminarycloud/mesh.py

@@ -106,8 +106,7 @@ class Mesh(ProtoWrapperBase):
             interval_seconds=interval_seconds,
             timeout_seconds=timeout_seconds,
         )
-        self._proto = get_mesh(self.id)._proto
-        return self.status
+        return self.refresh().status
 
     def refresh(self) -> "Mesh":
         """

luminarycloud/params/enum/_enum_wrappers.py

@@ -541,6 +541,31 @@ class MaterialFluidPreset(_IntEnum):
     CUSTOM_MATERIAL_FLUID = _clientpb.CUSTOM_MATERIAL_FLUID
 
 
+class ThermalConductivityModelSolid(_IntEnum):
+    """
+    Uniform or direction-dependent conductivity models.
+
+    Attributes
+    ----------
+    ISOTROPIC_CONDUCTIVITY
+        Equal thermal conductivity in all directions.
+    ORTHOTROPIC_CONDUCTIVITY
+        Different thermal conductivity along each axis of the frames to which
+                        the volumes of this material are assigned.
+
+
+    Examples
+    --------
+    >>> from luminarycloud.params.enum import ThermalConductivityModelSolid
+    >>> ThermalConductivityModelSolid.ISOTROPIC_CONDUCTIVITY
+    >>> ThermalConductivityModelSolid.ORTHOTROPIC_CONDUCTIVITY
+    """
+
+    INVALID = _clientpb.INVALID_THERMAL_CONDUCTIVITY_MODEL_SOLID
+    ISOTROPIC_CONDUCTIVITY = _clientpb.ISOTROPIC_CONDUCTIVITY
+    ORTHOTROPIC_CONDUCTIVITY = _clientpb.ORTHOTROPIC_CONDUCTIVITY
+
+
 class MaterialSolidPreset(_IntEnum):
     """
     Select a predefined set of material properties or allow a custom set of properties.

luminarycloud/params/simulation/material/material_solid_.py

@@ -25,9 +25,15 @@ class MaterialSolid(CodeRepr, ParamGroupWrapper[clientpb.MaterialSolid]):
     density: LcFloat = 2700
     "Constant density value."
     specific_heat_cp: LcFloat = 896
-    "Specific heat at constant pressure."
+    "Specific heat."
+    conductivity_model: enum.ThermalConductivityModelSolid = (
+        enum.ThermalConductivityModelSolid.ISOTROPIC_CONDUCTIVITY
+    )
+    "Uniform or direction-dependent conductivity models."
     thermal_conductivity: LcFloat = 167
     "The thermal conductivity of the material."
+    orthotropic_conductivity: Vector3 = field(default_factory=lambda: Vector3(167, 167, 167))
+    "The thermal conductivities of the material along each axis."
     thermal_conductivity_table: RectilinearTable | None = None
     "Correlation between thermal conductivity and temperature."
     preset: enum.MaterialSolidPreset = enum.MaterialSolidPreset.ALUMINUM
@@ -37,7 +43,11 @@ class MaterialSolid(CodeRepr, ParamGroupWrapper[clientpb.MaterialSolid]):
         _proto = clientpb.MaterialSolid()
         _proto.constant_density_value_solid.CopyFrom(_to_ad_proto(self.density))
         _proto.specific_heat_cp_solid.CopyFrom(_to_ad_proto(self.specific_heat_cp))
+        _proto.thermal_conductivity_model_solid = self.conductivity_model.value
         _proto.thermal_conductivity_constant_solid.CopyFrom(_to_ad_proto(self.thermal_conductivity))
+        _proto.thermal_conductivity_orthotropic_solid.CopyFrom(
+            self.orthotropic_conductivity._to_ad_proto()
+        )
         if self.thermal_conductivity_table is not None:
             if self.thermal_conductivity_table.table_type != _param_name_to_table_type(
                 "thermal_conductivity_table_data"
@@ -50,7 +60,11 @@ class MaterialSolid(CodeRepr, ParamGroupWrapper[clientpb.MaterialSolid]):
     def _from_proto(self, proto: clientpb.MaterialSolid) -> None:
         self.density = _from_ad_proto(proto.constant_density_value_solid)
         self.specific_heat_cp = _from_ad_proto(proto.specific_heat_cp_solid)
+        self.conductivity_model = enum.ThermalConductivityModelSolid(
+            proto.thermal_conductivity_model_solid
+        )
         self.thermal_conductivity = _from_ad_proto(proto.thermal_conductivity_constant_solid)
+        self.orthotropic_conductivity._from_ad_proto(proto.thermal_conductivity_orthotropic_solid)
         if proto.thermal_conductivity_table_data != "":
             self.thermal_conductivity_table = RectilinearTable(
                 id=proto.thermal_conductivity_table_data,

luminarycloud/physics_ai/architectures.py

@@ -1,5 +1,6 @@
 # Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
 from typing import List, Optional
+import json
 
 from .._client import get_default_client
 from .._proto.api.v0.luminarycloud.physics_ai import physics_ai_pb2 as physaipb
@@ -7,6 +8,8 @@ from .._wrapper import ProtoWrapper, ProtoWrapperBase
 from ..types.ids import PhysicsAiArchitectureID, PhysicsAiArchitectureVersionID
 from ..enum.physics_ai_lifecycle_state import PhysicsAiLifecycleState
 
+from .training_jobs import PhysicsAiTrainingJob
+
 
 @ProtoWrapper(physaipb.PhysicsAiArchitectureVersion)
 class PhysicsAiArchitectureVersion(ProtoWrapperBase):
@@ -18,10 +21,65 @@ class PhysicsAiArchitectureVersion(ProtoWrapperBase):
 
     id: PhysicsAiArchitectureVersionID
     name: str
+    architecture_name: str
     changelog: str
     lifecycle_state: PhysicsAiLifecycleState
     _proto: physaipb.PhysicsAiArchitectureVersion
 
+    def get_training_description(self, config: dict) -> str:
+        if config.get("description"):
+            return config["description"]
+
+        desc = f"Training job for architecture {self.name}"
+        if config.get("custom_args"):
+            desc += f" with custom args: {config['custom_args']}"
+        return desc
+
+    def train(self, config: dict) -> PhysicsAiTrainingJob:
+        """
+        Submit a training job for this architecture version.
+
+        Parameters
+        ----------
+        config : dict, optional
+            Training configuration dictionary
+
+        Returns
+        -------
+        PhysicsAiTrainingJob
+            The submitted training job object
+        """
+
+        if "custom_args" not in config:
+            config["custom_args"] = ""
+        if "priority_class" not in config:
+            config["priority_class"] = "prod-batch-priority"
+        if "resources" not in config:
+            config["resources"] = {}
+        if "process_gpus" not in config["resources"]:
+            config["resources"]["process_gpus"] = 8
+        if "train_gpus" not in config["resources"]:
+            config["resources"]["train_gpus"] = 8
+        if "test_gpus" not in config["resources"]:
+            config["resources"]["test_gpus"] = 8
+        if "mode" not in config:
+            config["mode"] = "full-gpu"
+
+        training_config_json = json.dumps(config, indent=2)
+        external_dataset_uri = f"gs://training-data/architecture-{self.id}"
+        req = physaipb.SubmitTrainingJobRequest(
+            architecture_version_id=self.id,
+            training_description=self.get_training_description(config),
+            external_dataset_uri=external_dataset_uri,
+            training_config=training_config_json,
+            initialization_type=physaipb.MODEL_INITIALIZATION_TYPE_RANDOM,
+            base_model_version_id="",
+        )
+
+        response = get_default_client().SubmitTrainingJob(req)
+
+        return PhysicsAiTrainingJob(response.training_job)
+
 
 @ProtoWrapper(physaipb.PhysicsAiArchitecture)
 class PhysicsAiArchitecture(ProtoWrapperBase):

luminarycloud/physics_ai/training_jobs.py

@@ -0,0 +1,37 @@
+# Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
+from typing import List, Optional
+from datetime import datetime
+
+from .._client import get_default_client
+from .._proto.api.v0.luminarycloud.physics_ai import physics_ai_pb2 as physaipb
+from .._proto.base import base_pb2 as basepb
+from .._wrapper import ProtoWrapper, ProtoWrapperBase
+from ..types.ids import PhysicsAiArchitectureVersionID
+
+
+@ProtoWrapper(physaipb.PhysicsAiTrainingJob)
+class PhysicsAiTrainingJob(ProtoWrapperBase):
+    """
+    Represents a Physics AI training job.
+
+    .. warning:: This feature is experimental and may change or be removed without notice.
+    """
+
+    id: str
+    architecture_version_id: PhysicsAiArchitectureVersionID
+    user_id: str
+    training_config: str
+    training_data_source_type: physaipb.TrainingDataSourceType
+    training_description: str
+    external_dataset_uri: str
+    initialization_type: physaipb.ModelInitializationType
+    base_model_version_id: str
+    status: basepb.JobStatus
+    error_message: str
+    output_model_version_id: str
+    creation_time: datetime
+    update_time: datetime
+    _proto: physaipb.PhysicsAiTrainingJob
+
+    def get_status(self) -> str:
+        return basepb.JobStatusType.Name(self.status.typ)

luminarycloud/pipelines/api.py

@@ -36,7 +36,6 @@ class PipelineRecord:
 class PipelineJobRecord:
     id: str
     pipeline_id: str
-    project_id: str
     name: str
     description: str | None
     status: Literal["pending", "running", "completed", "failed", "cancelled"]
@@ -50,7 +49,6 @@ class PipelineJobRecord:
         return cls(
             id=json["id"],
             pipeline_id=json["pipeline_id"],
-            project_id=json["project_id"],
             name=json["name"],
             description=json["description"],
             status=json["status"],
@@ -88,7 +86,7 @@ def create_pipeline(
         "description": description,
     }
     res = get_default_client().http.post("/rest/v0/pipelines", body)
-    return PipelineRecord.from_json(res)
+    return PipelineRecord.from_json(res["data"])
 
 
 def list_pipelines() -> list[PipelineRecord]:
@@ -96,7 +94,7 @@ def list_pipelines() -> list[PipelineRecord]:
     List all pipelines.
     """
     res = get_default_client().http.get("/rest/v0/pipelines")
-    return [PipelineRecord.from_json(p) for p in res]
+    return [PipelineRecord.from_json(p) for p in res["data"]]
 
 
 def get_pipeline(id: str) -> PipelineRecord:
@@ -109,11 +107,11 @@ def get_pipeline(id: str) -> PipelineRecord:
         ID of the pipeline to fetch.
     """
     res = get_default_client().http.get(f"/rest/v0/pipelines/{id}")
-    return PipelineRecord.from_json(res)
+    return PipelineRecord.from_json(res["data"])
 
 
 def create_pipeline_job(
-    pipeline_id: str, args: PipelineArgs, project_id: str, name: str, description: str | None = None
+    pipeline_id: str, args: PipelineArgs, name: str, description: str | None = None
 ) -> PipelineJobRecord:
     """
     Create a new pipeline job.
@@ -124,8 +122,6 @@ def create_pipeline_job(
         ID of the pipeline to invoke.
     args : PipelineArgs
         Arguments to pass to the pipeline.
-    project_id : str
-        ID of the project to run the pipeline job in.
     name : str
         Name of the pipeline job.
     description : str, optional
@@ -135,13 +131,13 @@ def create_pipeline_job(
     arg_rows = [row.row_values for row in args.rows]
     body = {
         "name": name,
-        "project_id": project_id,
+        "description": description,
         "argument_names": [p.name for p in args.params],
         "argument_rows": arg_rows,
     }
 
     res = get_default_client().http.post(f"/rest/v0/pipelines/{pipeline_id}/pipeline_jobs", body)
-    return PipelineJobRecord.from_json(res)
+    return PipelineJobRecord.from_json(res["data"])
 
 
 def get_pipeline_job(id: str) -> PipelineJobRecord:
@@ -149,7 +145,7 @@ def get_pipeline_job(id: str) -> PipelineJobRecord:
     Get a pipeline job by ID.
     """
     res = get_default_client().http.get(f"/rest/v0/pipeline_jobs/{id}")
-    return PipelineJobRecord.from_json(res)
+    return PipelineJobRecord.from_json(res["data"])
 
 
 def list_pipeline_jobs() -> list[PipelineJobRecord]:
@@ -157,4 +153,4 @@ def list_pipeline_jobs() -> list[PipelineJobRecord]:
     List all pipeline jobs.
     """
     res = get_default_client().http.get("/rest/v0/pipeline_jobs")
-    return [PipelineJobRecord.from_json(p) for p in res]
+    return [PipelineJobRecord.from_json(p) for p in res["data"]]

luminarycloud/simulation.py

@@ -57,6 +57,8 @@ class Simulation(ProtoWrapperBase):
     "ID of the simulation mesh."
     project_id: ProjectID
     "ID of the project containing this simulation."
+    doe_name: str
+    "Name of the design of experiments that created this simulation."
 
     _proto: simulationpb.Simulation
 
@@ -122,8 +124,7 @@ class Simulation(ProtoWrapperBase):
             interval_seconds=interval_seconds,
             timeout_seconds=timeout_seconds,
         )
-        self._proto = get_simulation(self.id)._proto
-        return self.status
+        return self.refresh().status
 
     def refresh(self) -> "Simulation":
         """

luminarycloud/simulation_template.py

@@ -490,7 +490,8 @@ class SimulationTemplate(ProtoWrapperBase):
         code += "\n\n\n"
         code += '# Create a new simulation template or modify the one that is synced with the UI "Setup" tab.\n'
         code += f'project = luminarycloud.get_project("{self.project_id}")\n'
-        code += f"template = project.create_simulation_template(name={self.name})\n"
+        escaped_name = self.name.replace('\\','\\\\').replace('"','\\"')
+        code += f'template = project.create_simulation_template(name="{escaped_name}")\n'
         code += '# TODO(USER): To modify the "Setup" template, uncomment the line below and comment out the line above.\n'
         code += "# template = project.list_simulation_templates()[0] # Setup template\n\n"
 

luminarycloud/vis/__init__.py

@@ -4,8 +4,12 @@ from .visualization import (
     EntityType as EntityType,
     list_renders as list_renders,
     list_quantities as list_quantities,
+    list_quantities as list_quantities,
+    list_cameras as list_cameras,
+    get_camera as get_camera,
     DirectionalCamera as DirectionalCamera,
     LookAtCamera as LookAtCamera,
+    CameraEntry as CameraEntry,
 )
 
 from .primitives import (
@@ -52,3 +56,14 @@ from .interactive_scene import (
 from .interactive_inference import (
     InteractiveInference as InteractiveInference,
 )
+
+# Unreleased/internal for testing now
+
+# from .report import (
+#    Report as Report,
+# )
+
+# from .interactive_report import (
+#    InteractiveReport as InteractiveReport,
+#    ReportEntry as ReportEntry,
+# )