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

luminarycloud/params/simulation/adaptive_mesh_refinement_.py

@@ -41,6 +41,8 @@ class AdaptiveMeshRefinement(CodeRepr, ParamGroupWrapper[clientpb.AdaptiveMeshRe
     "The method to generate the computational mesh."
     target_cv_millions: int = 10
     "User-requested mesh size in millions of control volumes."
+    all_tet: enum.AllTet = enum.AllTet.ALL_TET_ON
+    "Automatically inserts high aspect ratio mesh elements in the boundary layer and ignores all adaptation boundary layer settings."
     user_scaling: LcFloat = 1.0
     "Scale factor between the geometry and the mesh."
     boundary_layer_profile: list[BoundaryLayerProfile] = field(default_factory=list)
@@ -55,6 +57,7 @@ class AdaptiveMeshRefinement(CodeRepr, ParamGroupWrapper[clientpb.AdaptiveMeshRe
         _proto.final_target_complexity.CopyFrom(_to_ad_proto(self.final_target_complexity))
         _proto.meshing_method = self.meshing_method.value
         _proto.target_cv_millions.value = self.target_cv_millions
+        _proto.all_tet = self.all_tet.value
         _proto.user_scaling.CopyFrom(_to_ad_proto(self.user_scaling))
         if self.boundary_layer_profile is not None:
             _proto.boundary_layer_profile.extend(v._to_proto() for v in self.boundary_layer_profile)
@@ -68,6 +71,7 @@ class AdaptiveMeshRefinement(CodeRepr, ParamGroupWrapper[clientpb.AdaptiveMeshRe
         self.final_target_complexity = _from_ad_proto(proto.final_target_complexity)
         self.meshing_method = enum.MeshingMethod(proto.meshing_method)
         self.target_cv_millions = proto.target_cv_millions.value
+        self.all_tet = enum.AllTet(proto.all_tet)
         self.user_scaling = _from_ad_proto(proto.user_scaling)
         self.boundary_layer_profile = [
             BoundaryLayerProfile.from_proto(v) for v in proto.boundary_layer_profile

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/params/simulation/physics/__init__.py

@@ -1,6 +1,5 @@
 from . import fluid
 from . import heat
-from . import periodic_pair
 from . import solution_controls
 from .fluid_ import Fluid
 from .heat_ import Heat

luminarycloud/params/simulation/physics/periodic_pair_.py

@@ -17,17 +17,6 @@ import luminarycloud.params.enum._enum_wrappers as enum
 
 from luminarycloud.params.simulation._lib import ParamGroupWrapper, create_unique_id
 
-from luminarycloud.params.simulation.physics.periodic_pair.periodicity_type_ import PeriodicityType
-from luminarycloud.params.simulation.physics.periodic_pair.periodicity_type_ import *
-from luminarycloud.params.simulation.physics.periodic_pair.periodicity_type.rotational_periodicity_ import (
-    RotationalPeriodicity,
-)
-from luminarycloud.params.simulation.physics.periodic_pair.periodicity_type.rotational_periodicity_ import *
-from luminarycloud.params.simulation.physics.periodic_pair.periodicity_type.translational_periodicity_ import (
-    TranslationalPeriodicity,
-)
-from luminarycloud.params.simulation.physics.periodic_pair.periodicity_type.translational_periodicity_ import *
-
 
 @dataclass(kw_only=True)
 class PeriodicPair(CodeRepr, ParamGroupWrapper[clientpb.PeriodicPair]):
@@ -39,8 +28,12 @@ class PeriodicPair(CodeRepr, ParamGroupWrapper[clientpb.PeriodicPair]):
     ""
     surfaces_side_b: list[str] = field(default_factory=list)
     ""
-    periodicity_type: PeriodicityType = field(default_factory=TranslationalPeriodicity)
-    " Possible types: ``TranslationalPeriodicity``, ``RotationalPeriodicity`` from the ``periodicity_type`` module."
+    translation: Vector3 = field(default_factory=lambda: Vector3(0.0, 0.0, 0.0))
+    "Holds the translation vector in the x,y,z directions from each point on the first periodic boundary to its matching point on the second periodic boundary."
+    center: Vector3 = field(default_factory=lambda: Vector3(0.0, 0.0, 0.0))
+    "Origin for the rotational periodicity transformation."
+    rotation_vector: Vector3 = field(default_factory=lambda: Vector3(0.0, 0.0, 0.0))
+    "Holds the rotation vector needed to transform each point on the first periodic boundary to its matching point on the second periodic boundary. The vector direction defines the periodicity axis and its magnitude defines the periodicity angle, in degrees, from A to B according to the right-hand rule."
 
     def _to_proto(self) -> clientpb.PeriodicPair:
         _proto = clientpb.PeriodicPair()
@@ -50,28 +43,16 @@ class PeriodicPair(CodeRepr, ParamGroupWrapper[clientpb.PeriodicPair]):
             _proto.bound_a.extend(self.surfaces_side_a)
         if self.surfaces_side_b is not None:
             _proto.bound_b.extend(self.surfaces_side_b)
-        if isinstance(self.periodicity_type, TranslationalPeriodicity):
-            _proto.periodic_bc_type = clientpb.TRANSLATIONAL
-            _proto.periodic_translation.CopyFrom(self.periodicity_type.translation._to_ad_proto())
-        if isinstance(self.periodicity_type, RotationalPeriodicity):
-            _proto.periodic_bc_type = clientpb.ROTATIONAL
-            _proto.periodic_center_of_rotation.CopyFrom(self.periodicity_type.center._to_ad_proto())
-            _proto.periodic_rotation_angles.CopyFrom(
-                self.periodicity_type.rotation_vector._to_ad_proto()
-            )
+        _proto.periodic_translation.CopyFrom(self.translation._to_ad_proto())
+        _proto.periodic_center_of_rotation.CopyFrom(self.center._to_ad_proto())
+        _proto.periodic_rotation_angles.CopyFrom(self.rotation_vector._to_ad_proto())
         return _proto
 
     def _from_proto(self, proto: clientpb.PeriodicPair) -> None:
         self.name = proto.periodic_pair_name
         self.surfaces_side_a.extend(proto.bound_a)
         self.surfaces_side_b.extend(proto.bound_b)
-        if proto.periodic_bc_type == clientpb.INVALID_PERIODIC_BC_TYPE:
-            raise ValueError("Invalid periodicity_type")
-        elif proto.periodic_bc_type == clientpb.TRANSLATIONAL:
-            self.periodicity_type = TranslationalPeriodicity()
-            self.periodicity_type.translation._from_ad_proto(proto.periodic_translation)
-        elif proto.periodic_bc_type == clientpb.ROTATIONAL:
-            self.periodicity_type = RotationalPeriodicity()
-            self.periodicity_type.center._from_ad_proto(proto.periodic_center_of_rotation)
-            self.periodicity_type.rotation_vector._from_ad_proto(proto.periodic_rotation_angles)
+        self.translation._from_ad_proto(proto.periodic_translation)
+        self.center._from_ad_proto(proto.periodic_center_of_rotation)
+        self.rotation_vector._from_ad_proto(proto.periodic_rotation_angles)
         return None

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_cpus" not in config["resources"]:
+            config["resources"]["process_cpus"] = 8
+        if "train_gpus" not in config["resources"]:
+            config["resources"]["train_gpus"] = 1
+        if "test_gpus" not in config["resources"]:
+            config["resources"]["test_gpus"] = 1
+        if "mode" not in config:
+            config["mode"] = "full"
+
+        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/inference.py

@@ -17,6 +17,7 @@ from ..project import Project
 from ..project import Project
 from .._helpers import upload_file
 from .._proto.upload import upload_pb2 as uploadpb
+from ..types.ids import PhysicsAiModelVersionID
 
 
 @dataclass
@@ -50,7 +51,7 @@ class ExtAeroInferenceResult:
 def external_aero_inference(
     project: Project,
     stl_file: str,
-    artifact_url: str,
+    model_version_id: PhysicsAiModelVersionID,
     conditions: Optional[Dict[str, Any]] = None,
     settings: Optional[Dict[str, Any]] = None,
     write_visualization_data=False,
@@ -62,8 +63,8 @@ def external_aero_inference(
         The project to which the inference files will be added.
     stl_file : str
         Fullpath the STL file to be used for inference.
-    artifact_url : str
-        Fullpath of the model artifact directory to be used for inference.
+    model_version_id : PhysicsAiModelVersionID
+        The ID of the trained model version to use for inference.
     conditions : Dict[str, Any], optional
         Dictionary of conditions to be passed to the inference service (e.g., alpha, beta, etc.).
     settings : Dict[str, Any], optional
@@ -80,7 +81,7 @@ def external_aero_inference(
     """
 
     result = perform_inference(
-        project, stl_file, artifact_url, conditions, settings, write_visualization_data
+        project, stl_file, model_version_id, conditions, settings, write_visualization_data
     )
     return ExtAeroInferenceResult(result)
 
@@ -88,7 +89,7 @@ def external_aero_inference(
 def perform_inference(
     project: Project,
     stl_file: str,
-    artifact_url: str,
+    model_version_id: PhysicsAiModelVersionID,
     conditions: Optional[Dict[str, Any]] = None,
     settings: Optional[Dict[str, Any]] = None,
     write_visualization_data=False,
@@ -100,8 +101,8 @@ def perform_inference(
         The project to which the inference files will be added.
     stl_file : str
         Fullpath the STL file to be used for inference.
-    artifact_url : str
-        Fullpath of the model artifact directory to be used for inference.
+    model_version_id : PhysicsAiModelVersionID
+        The ID of the trained model version to use for inference.
     conditions : Dict[str, Any], optional
         Dictionary of conditions to be passed to the inference service (e.g., alpha, beta, etc.).
     settings : Dict[str, Any], optional
@@ -142,7 +143,7 @@ def perform_inference(
     stl_url = upload_if_file(stl_file)
 
     raw = start_inference_job(
-        project, stl_url, artifact_url, conditions, settings, write_visualization_data
+        project, stl_url, model_version_id, conditions, settings, write_visualization_data
     )
     currated: dict[str, Any] = {}
     for k, v in raw.items():
@@ -162,7 +163,7 @@ def perform_inference(
 def start_inference_job(
     project: Project,
     stl_url: str,
-    artifact_url: str,
+    model_version_id: PhysicsAiModelVersionID,
     conditions: Optional[Dict[str, Any]] = None,
     settings: Optional[Dict[str, Any]] = None,
     write_visualization_data=False,
@@ -174,8 +175,8 @@ def start_inference_job(
         Reference to a project.
     stl_url : str
         URL of the STL file to be used for inference.
-    artifact_url : str
-        URL of the model artifact directory to be used for inference.
+    model_version_id : PhysicsAiModelVersionID
+        The ID of the trained model version to use for inference.
     conditions : Dict[str, Any], optional
         Dictionary of conditions to be passed to the inference service (e.g., alpha, beta, etc.).
     settings : Dict[str, Any], optional
@@ -203,13 +204,12 @@ def start_inference_job(
 
     req = inferencepb.CreateInferenceServiceJobRequest(
         stl_url=stl_url,
-        artifact_url=artifact_url,
+        model_version_id=str(model_version_id),
         conditions=conditions_bytes,
         settings=settings_bytes,
         project_id=project.id,
         write_visualization_data=write_visualization_data,
     )
-
     res: inferencepb.CreateInferenceServiceJobResponse = (
         get_default_client().CreateInferenceServiceJob(req)
     )

luminarycloud/physics_ai/solution.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import tarfile
-from typing import List, Optional, BinaryIO, cast
+from typing import List, Optional, BinaryIO, cast, Dict
 
 from .._client import get_default_client
 from .._helpers.download import download_solution_physics_ai as _download_solution_physics_ai
@@ -16,6 +16,7 @@ def _download_processed_solution_physics_ai(  # noqa: F841
     volume_fields_to_keep: Optional[List[QuantityType]] = None,
     process_volume: bool = False,
     single_precision: bool = True,
+    internal_options: Optional[Dict[str, str]] = None,
 ) -> tarfile.TarFile:
     """
     Download solution data with physics AI processing applied.
@@ -50,6 +51,7 @@ def _download_processed_solution_physics_ai(  # noqa: F841
         volume_fields_to_keep=volume_fields_to_keep,
         process_volume=process_volume,
         single_precision=single_precision,
+        internal_options=internal_options,
     )
 
     assert stream is not None, "Failed to download solution data"

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/__init__.py

@@ -2,6 +2,8 @@
 from .core import (
     Pipeline as Pipeline,
     PipelineParameter as PipelineParameter,
+    # Stage base class, mainly exported for testing
+    Stage as Stage,
 )
 
 from .parameters import (
@@ -11,9 +13,7 @@ from .parameters import (
     BoolPipelineParameter as BoolPipelineParameter,
 )
 
-from .operators import (
-    # Operator base class, mainly exported for testing
-    Operator as Operator,
+from .stages import (
     # PipelineOutputs, i.e. things that "flow" in a Pipeline
     PipelineOutputGeometry as PipelineOutputGeometry,
     PipelineOutputMesh as PipelineOutputMesh,
@@ -21,6 +21,8 @@ from .operators import (
     # Concrete operators and their output types
     ReadGeometry as ReadGeometry,
     ReadGeometryOutputs as ReadGeometryOutputs,
+    ReadMesh as ReadMesh,
+    ReadMeshOutputs as ReadMeshOutputs,
     ModifyGeometry as ModifyGeometry,
     ModifyGeometryOutputs as ModifyGeometryOutputs,
     Mesh as Mesh,
@@ -39,4 +41,10 @@ from .api import (
     list_pipelines as list_pipelines,
     get_pipeline as get_pipeline,
     create_pipeline_job as create_pipeline_job,
+    get_pipeline_job as get_pipeline_job,
+    list_pipeline_jobs as list_pipeline_jobs,
+    PipelineJobRecord as PipelineJobRecord,
+    PipelineRecord as PipelineRecord,
+    PipelineJobRunRecord as PipelineJobRunRecord,
+    LogLine as LogLine,
 )