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

luminarycloud/_proto/physicsaiinferenceservice/physicsaiinferenceservice_pb2.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: proto/physicsaiinferenceservice/physicsaiinferenceservice.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.base import base_pb2 as proto_dot_base_dot_base__pb2
+from luminarycloud._proto.api.v0.luminarycloud.physicsaiinference import physicsaiinference_pb2 as proto_dot_api_dot_v0_dot_luminarycloud_dot_physicsaiinference_dot_physicsaiinference__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')
+
+
+
+_PHYSICSAIINFERENCESERVICE = DESCRIPTOR.services_by_name['PhysicsAiInferenceService']
+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
+# @@protoc_insertion_point(module_scope)

luminarycloud/_proto/physicsaiinferenceservice/physicsaiinferenceservice_pb2.pyi

@@ -0,0 +1,7 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import google.protobuf.descriptor
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2.py

@@ -15,7 +15,7 @@ _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')
+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')
 
 
 
@@ -25,5 +25,5 @@ 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
+  _PHYSICSAITRAININGSERVICE._serialized_end=535
 # @@protoc_insertion_point(module_scope)

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2_grpc.py

@@ -20,6 +20,11 @@ class PhysicsAiTrainingServiceStub(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.physicsaitrainingservice.PhysicsAiTrainingService/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,
+                )
 
 
 class PhysicsAiTrainingServiceServicer(object):
@@ -33,6 +38,13 @@ class PhysicsAiTrainingServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CancelTrainingJob(self, request, context):
+        """Cancel 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 = {
@@ -41,6 +53,11 @@ def add_PhysicsAiTrainingServiceServicer_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,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'luminary.proto.physicsaitrainingservice.PhysicsAiTrainingService', rpc_method_handlers)
@@ -68,3 +85,20 @@ class PhysicsAiTrainingService(object):
             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)
+
+    @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.physicsaitrainingservice.PhysicsAiTrainingService/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)

luminarycloud/_proto/physicsaitrainingservice/physicsaitrainingservice_pb2_grpc.pyi

@@ -15,6 +15,11 @@ class PhysicsAiTrainingServiceStub:
         luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse,
     ]
     """Submit a physics AI training job"""
+    CancelTrainingJob: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobRequest,
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobResponse,
+    ]
+    """Cancel a physics AI training job"""
 
 class PhysicsAiTrainingServiceServicer(metaclass=abc.ABCMeta):
     """PhysicsAiTrainingService provides training functionality for Physics AI"""
@@ -26,5 +31,12 @@ class PhysicsAiTrainingServiceServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse:
         """Submit a physics AI training job"""
+    @abc.abstractmethod
+    def CancelTrainingJob(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobResponse:
+        """Cancel a physics AI training job"""
 
 def add_PhysicsAiTrainingServiceServicer_to_server(servicer: PhysicsAiTrainingServiceServicer, server: grpc.Server) -> None: ...

luminarycloud/enum/vis_enums.py

@@ -73,6 +73,12 @@ class VisQuantity(IntEnum):
     Q_CRITERION_TIME_AVERAGE = quantitypb.Q_CRITERION_TIME_AVERAGE
     HEAT_FLUX_TIME_AVERAGE = quantitypb.HEAT_FLUX_TIME_AVERAGE
     DEBUG_QUANTITY = quantitypb.DEBUG_QUANTITY
+    # Actuator disk quanties
+    THRUST_PER_UNIT_AREA = quantitypb.THRUST_PER_UNIT_AREA
+    TORQUE_PER_UNIT_AREA = quantitypb.TORQUE_PER_UNIT_AREA
+    BLADE_LOCAL_ANGLE_OF_ATTACK = quantitypb.BLADE_LOCAL_ANGLE_OF_ATTACK
+    BLADE_SECTIONAL_DRAG_COEFFICIENT = quantitypb.BLADE_SECTIONAL_DRAG_COEFFICIENT
+    BLADE_SECTIONAL_LIFT_COEFFICIENT = quantitypb.BLADE_SECTIONAL_LIFT_COEFFICIENT
 
 
 # Return the text name for the VisQuantity including the units, as it appears in the UI

luminarycloud/geometry.py

@@ -61,6 +61,10 @@ class Geometry(ProtoWrapperBase):
         """
         return timestamp_to_datetime(self._proto.update_time)
 
+    @property
+    def url(self) -> str:
+        return f"{self.project().url}/geometry/{self.id}"
+
     def project(self) -> "Project":
         """
         Get the project this geometry belongs to.

luminarycloud/geometry_version.py

@@ -31,6 +31,10 @@ class GeometryVersion(ProtoWrapperBase):
         """
         return timestamp_to_datetime(self._proto.create_time)
 
+    @property
+    def url(self) -> str:
+        return f"{self.geometry().url}/version/{self.id}"
+
     def geometry(self) -> Geometry:
         """
         Get the parent geometry.

luminarycloud/mesh.py

@@ -38,6 +38,10 @@ class Mesh(ProtoWrapperBase):
     def create_time(self) -> datetime:
         return timestamp_to_datetime(self._proto.create_time)
 
+    @property
+    def url(self) -> str:
+        return f"{self.project().url}/mesh/{self.id}"
+
     def project(self) -> "Project":
         """
         Get the project this mesh belongs to.

luminarycloud/meshing/mesh_generation_params.py

@@ -16,11 +16,10 @@ from ..params.geometry import (
 )
 from ..types import Vector3
 from .sizing_strategy import MaxCount, Minimal, MinimalCount, SizingStrategy, TargetCount
-from ..pipeline_util.dictable import PipelineDictable
 
 
 @dataclass(kw_only=True)
-class VolumeMeshingParams(PipelineDictable):
+class VolumeMeshingParams:
     """Volume meshing parameters."""
 
     volumes: list[Volume]
@@ -39,7 +38,7 @@ class VolumeMeshingParams(PipelineDictable):
 
 
 @dataclass(kw_only=True)
-class ModelMeshingParams(PipelineDictable):
+class ModelMeshingParams:
     """Model meshing parameters."""
 
     surfaces: Sequence[Surface | str]
@@ -62,7 +61,7 @@ class ModelMeshingParams(PipelineDictable):
 
 
 @dataclass(kw_only=True)
-class BoundaryLayerParams(PipelineDictable):
+class BoundaryLayerParams:
     """Boundary layer meshing parameters."""
 
     surfaces: Sequence[Surface | str]
@@ -88,7 +87,7 @@ class BoundaryLayerParams(PipelineDictable):
 
 
 @dataclass(kw_only=True)
-class RefinementRegion(PipelineDictable):
+class RefinementRegion:
     """Refinement region parameters."""
 
     name: str
@@ -138,7 +137,7 @@ class RefinementRegion(PipelineDictable):
 
 
 @dataclass(kw_only=True)
-class MeshGenerationParams(PipelineDictable):
+class MeshGenerationParams:
     """Mesh generation parameters."""
 
     geometry_id: str

luminarycloud/meshing/sizing_strategy/sizing_strategies.py

@@ -1,11 +1,10 @@
 from dataclasses import dataclass
 
 from luminarycloud._helpers.warnings.deprecated import deprecated
-from ...pipeline_util.dictable import PipelineDictable
 
 
 @dataclass
-class SizingStrategy(PipelineDictable):
+class SizingStrategy:
     """Sizing strategy parameters."""
 
     pass

luminarycloud/physics_ai/solution.py

@@ -17,6 +17,7 @@ def _download_processed_solution_physics_ai(  # noqa: F841
     process_volume: bool = False,
     single_precision: bool = True,
     internal_options: Optional[Dict[str, str]] = None,
+    export_surface_groups: Optional[Dict[str, List[str]]] = None,
 ) -> tarfile.TarFile:
     """
     Download solution data with physics AI processing applied.
@@ -37,6 +38,8 @@ def _download_processed_solution_physics_ai(  # noqa: F841
             If None, all available volume fields are included.
         process_volume: Whether to process volume data
         single_precision: Whether to use single precision for floating point fields
+        export_surface_groups: Dictionary mapping group names to lists of surface names.
+            Each group will be exported as an individual STL file.
 
     Raises:
         ValueError: If invalid field names are provided
@@ -46,6 +49,7 @@ def _download_processed_solution_physics_ai(  # noqa: F841
         get_default_client(),
         solution_id,
         exclude_surfaces=exclude_surfaces,
+        export_surface_groups=export_surface_groups,
         fill_holes=fill_holes,
         surface_fields_to_keep=surface_fields_to_keep,
         volume_fields_to_keep=volume_fields_to_keep,

luminarycloud/pipelines/api.py

@@ -7,6 +7,7 @@ from time import time, sleep
 import logging
 
 from .arguments import PipelineArgValueType
+from .core import Stage
 from ..pipelines import Pipeline, PipelineArgs
 from .._client import get_default_client
 
@@ -69,6 +70,25 @@ class PipelineRecord:
         res = get_default_client().http.get(f"/rest/v0/pipelines/{self.id}/pipeline_jobs")
         return [PipelineJobRecord.from_json(p) for p in res["data"]]
 
+    def delete(self) -> None:
+        """
+        Delete this pipeline.
+
+        This will permanently delete the pipeline and all associated pipeline jobs.
+        This operation cannot be undone.
+
+        Raises
+        ------
+        HTTPException
+            If the pipeline does not exist or if you do not have permission to delete it.
+
+        Examples
+        --------
+        >>> pipeline = pipelines.get_pipeline("pipeline-123")
+        >>> pipeline.delete()
+        """
+        get_default_client().http.delete(f"/rest/v0/pipelines/{self.id}")
+
 
 @dataclass
 class PipelineJobRecord:
@@ -80,7 +100,7 @@ class PipelineJobRecord:
     pipeline_id: str
     name: str
     description: str | None
-    status: Literal["pending", "running", "completed", "failed"]
+    status: Literal["pending", "running", "completed", "failed", "cancelled"]
     created_at: datetime
     updated_at: datetime
     started_at: datetime | None
@@ -156,18 +176,37 @@ class PipelineJobRecord:
         res = get_default_client().http.get(f"/rest/v0/pipeline_jobs/{self.id}/artifacts")
         return res["data"]
 
+    def delete(self) -> None:
+        """
+        Delete this pipeline job.
+
+        This will permanently delete the pipeline job and all associated runs and tasks.
+        This operation cannot be undone.
+
+        Raises
+        ------
+        HTTPException
+            If the pipeline job does not exist or if you do not have permission to delete it.
+
+        Examples
+        --------
+        >>> pipeline_job = pipelines.get_pipeline_job("pipelinejob-123")
+        >>> pipeline_job.delete()
+        """
+        get_default_client().http.delete(f"/rest/v0/pipeline_jobs/{self.id}")
+
     def wait(
         self,
         *,
         interval_seconds: float = 5,
         timeout_seconds: float = float("inf"),
         print_logs: bool = False,
-    ) -> Literal["completed", "failed"]:
+    ) -> Literal["completed", "failed", "cancelled"]:
         """
-        Wait for the pipeline job to complete or fail.
+        Wait for the pipeline job to complete, fail, or be cancelled.
 
         This method polls the pipeline job status at regular intervals until it reaches
-        a terminal state (completed or failed).
+        a terminal state (completed, failed, or cancelled).
 
         Parameters
         ----------
@@ -180,7 +219,7 @@ class PipelineJobRecord:
 
         Returns
         -------
-        Literal["completed", "failed"]
+        Literal["completed", "failed", "cancelled"]
             The final status of the pipeline job.
 
         Raises
@@ -216,6 +255,9 @@ class PipelineJobRecord:
             elif updated_job.status == "failed":
                 logger.warning(f"Pipeline job {self.id} failed")
                 return "failed"
+            elif updated_job.status == "cancelled":
+                logger.info(f"Pipeline job {self.id} was cancelled")
+                return "cancelled"
 
             # Check timeout
             if time() >= deadline:
@@ -233,13 +275,53 @@ class PipelineJobRecord:
             self.started_at = updated_job.started_at
             self.completed_at = updated_job.completed_at
 
+    def get_concurrency_limits(self) -> dict[str, int]:
+        """
+        Returns the concurrency limits for this pipeline job.
+
+        Returns
+        -------
+        dict[str, int]
+            A dictionary mapping stage IDs to their concurrency limits.
+        """
+        res = get_default_client().http.get(f"/rest/v0/pipeline_jobs/{self.id}/concurrency_limits")
+        return {k: v["limit"] for k, v in res["data"].items()}
+
+    def set_concurrency_limits(self, limits: dict[str, int]) -> None:
+        """
+        Sets the concurrency limits for this pipeline job.
+
+        Parameters
+        ----------
+        limits : dict[str, int]
+            A dictionary mapping stage IDs to their concurrency limits.
+        """
+        body = {k: {"limit": v} for k, v in limits.items()}
+        get_default_client().http.put(f"/rest/v0/pipeline_jobs/{self.id}/concurrency_limits", body)
+
+    def cancel(self) -> None:
+        """Cancel this running pipeline job.
+
+        This will request cancellation of the underlying Prefect flow run. The
+        job should eventually transition to a cancelled terminal state once
+        the backend processes the cancellation.
+
+        Raises
+        ------
+        HTTPError
+            If the pipeline job cannot be cancelled (e.g., not found, not
+            running, or lacks the necessary Prefect flow run ID).
+        """
+        get_default_client().http.post(f"/rest/v0/pipeline_jobs/{self.id}/cancel", {})
+        logger.info(f"Cancelled pipeline job {self.id}")
+
 
 @dataclass
 class PipelineJobRunRecord:
     pipeline_job_id: str
     idx: int
     arguments: list[PipelineArgValueType]
-    status: Literal["pending", "running", "completed", "failed"]
+    status: Literal["pending", "running", "completed", "failed", "cancelled"]
 
     @classmethod
     def from_json(cls, json: dict) -> "PipelineJobRunRecord":
@@ -347,7 +429,11 @@ def get_pipeline(id: str) -> PipelineRecord:
 
 
 def create_pipeline_job(
-    pipeline_id: str, args: PipelineArgs, name: str, description: str | None = None
+    pipeline_id: str,
+    args: PipelineArgs,
+    name: str,
+    description: str | None = None,
+    concurrency_limits: dict[str, int] | None = None,
 ) -> PipelineJobRecord:
     """
     Create a new pipeline job.
@@ -362,6 +448,8 @@ def create_pipeline_job(
         Name of the pipeline job.
     description : str, optional
         Description of the pipeline job.
+    concurrency_limits : dict[str, int], optional
+        A dictionary mapping stage IDs to their concurrency limits.
     """
 
     arg_rows = [row.row_values for row in args.rows]
@@ -373,7 +461,10 @@ def create_pipeline_job(
     }
 
     res = get_default_client().http.post(f"/rest/v0/pipelines/{pipeline_id}/pipeline_jobs", body)
-    return PipelineJobRecord.from_json(res["data"])
+    pjr = PipelineJobRecord.from_json(res["data"])
+    if concurrency_limits is not None:
+        pjr.set_concurrency_limits(concurrency_limits)
+    return pjr
 
 
 def get_pipeline_job(id: str) -> PipelineJobRecord:

luminarycloud/pipelines/core.py

@@ -292,7 +292,7 @@ class Pipeline:
     def pipeline_params(self) -> set[PipelineParameter]:
         return self._stages_dict_and_params()[1]
 
-    def _get_stage_id(self, stage: Stage) -> str:
+    def get_stage_id(self, stage: Stage) -> str:
         return self._stage_ids[stage]
 
     def _stages_dict_and_params(self) -> tuple[dict, set[PipelineParameter]]:

luminarycloud/pipelines/stages.py

@@ -2,7 +2,7 @@
 from dataclasses import dataclass
 
 from .core import Stage, StageInputs, StageOutputs, PipelineOutput
-from .parameters import StringPipelineParameter, IntPipelineParameter
+from .parameters import BoolPipelineParameter, StringPipelineParameter, IntPipelineParameter
 from ..meshing import MeshGenerationParams
 
 
@@ -49,6 +49,20 @@ class ReadGeometry(Stage[ReadGeometryOutputs]):
     ----------
     geometry_id : str | StringPipelineParameter
         The ID of the Geometry to retrieve.
+    use_geo_without_copying : bool | BoolPipelineParameter
+        By default, this is False, meaning that each Geometry this stage references will be copied
+        and the PipelineJob will actually operate on the copied Geometry. This is done so that a
+        PipelineJob can be based on a single parametric Geometry which each PipelineJobRun modifies
+        by applying a NamedVariableSet. That modification mutates the Geometry, so those runs can
+        only happen in parallel without intefrering with each other if they each operate on a
+        different copy of the Geometry.
+
+        However, if you've already prepared your Geometry in advance and you don't want the
+        PipelineJob to create copies, you can set this to True. In that case, the referenced
+        Geometry will be used directly without being copied.
+
+        IMPORTANT: If you set this to True, you must ensure no two PipelineJobRuns operate on the
+        same Geometry, i.e. no two PipelineArgs rows contain the same Geometry ID.
 
     Outputs
     -------
@@ -61,10 +75,11 @@ class ReadGeometry(Stage[ReadGeometryOutputs]):
         *,
         stage_name: str | None = None,
         geometry_id: str | StringPipelineParameter,
+        use_geo_without_copying: bool | BoolPipelineParameter = False,
     ):
         super().__init__(
             stage_name,
-            {"geometry_id": geometry_id},
+            {"geometry_id": geometry_id, "use_geo_without_copying": use_geo_without_copying},
             StageInputs(self),
             ReadGeometryOutputs._instantiate_for(self),
         )
@@ -202,13 +217,6 @@ class Mesh(Stage[MeshOutputs]):
             MeshOutputs._instantiate_for(self),
         )
 
-    # TODO: bring back the full MeshGenerationParams, but we need to be able to hydrate it from the
-    # pipeline YAML. I can probably bake that logic into PipelineDictable, `from_pipeline_dict` or
-    # something.
-    # @classmethod
-    # def _parse_params(cls, params: dict) -> dict:
-    #     return {"mesh_gen_params": MeshGenerationParams.from_pipeline_dict(**params["mesh_gen_params"])}
-
 
 @dataclass
 class SimulateOutputs(StageOutputs):
@@ -230,6 +238,9 @@ class Simulate(Stage[SimulateOutputs]):
         The name to assign to the Simulation. If None, a default name will be used.
     sim_template_id : str | StringPipelineParameter
         The ID of the SimulationTemplate to use for the Simulation.
+    batch_processing : bool | BoolPipelineParameter, default True
+        If True, the Simulation will run as a standard job. If False, the Simulation will run as a
+        priority job.
 
     Outputs
     -------
@@ -244,10 +255,12 @@ class Simulate(Stage[SimulateOutputs]):
         mesh: PipelineOutputMesh,
         sim_name: str | StringPipelineParameter | None = None,
         sim_template_id: str | StringPipelineParameter,
+        batch_processing: bool | BoolPipelineParameter = True,
     ):
         super().__init__(
             stage_name,
             {
+                "batch_processing": batch_processing,
                 "sim_name": sim_name,
                 "sim_template_id": sim_template_id,
             },

luminarycloud/project.py

@@ -485,15 +485,14 @@ class Project(ProtoWrapperBase):
         description : str, optional
             Simulation description.
         batch_processing : bool, default True
-            If True, batch processing will be used for this
-            simulation.
-            Use Batch Processing on simulations that are not time-sensitive to
-            save up to 65% in credits.
+            If True, this simulation will run as a standard job. If False, this simulation will run
+            as a priority job.
         gpu_type : GPUType, optional
             GPU type to use for the simulation.
         gpu_count : int, optional
-            Number of GPUs to use for the simulation. Must be specified to a
-            positive value if `gpu_type` is specified.
+            Number of GPUs to use for the simulation. Only relevant if `gpu_type` is
+            specified. If this is set to 0 or omitted and `gpu_type` is specified, the number
+            of gpus will be automatically determined.
         """
 
         named_variable_set_version_id: Optional[str] = None

luminarycloud/types/vector3.py

@@ -8,11 +8,10 @@ from .adfloat import (
     _to_ad_proto as _float_to_ad_proto,
     _from_ad_proto as _float_from_ad_proto,
 )
-from ..pipeline_util.dictable import PipelineDictable
 
 
 @dataclass
-class Vector3(PipelineDictable):
+class Vector3:
     """Represents a 3-dimensional vector.
 
     Supports direct component access, indexing, iteration, and conversion to numpy arrays.

luminarycloud/vis/data_extraction.py

@@ -602,7 +602,7 @@ class DataExtractor:
         code += f"    if sol.id == '{self._solution.id}':\n"
         code += f"        solution = sol\n"
         code += f"        break\n"
-        code += "data_extractor = vis.DataExtractor(solution)\n"
+        code += f"{obj_name} = vis.DataExtractor(solution)\n"
         code += "\n"
 
         code += "\n"
@@ -615,11 +615,11 @@ class DataExtractor:
         for extract in self._extracts:
             # Name objects numerically: slice0, slice1, etc.
             name = _data_extract_to_obj_name(extract)
-            obj_name = f"{name}{name_map[obj_name]}"
-            name_map[obj_name] += 1
-            ids_to_obj_name[extract.id] = obj_name
-            code += extract._to_code_helper(obj_name, hide_defaults=hide_defaults)
-            code += f"data_extractor.add_data_extract({obj_name})\n"
+            extract_obj_name = f"{name}{name_map[name]}"
+            name_map[name] += 1
+            ids_to_obj_name[extract.id] = extract_obj_name
+            code += extract._to_code_helper(extract_obj_name, hide_defaults=hide_defaults)
+            code += f"{obj_name}.add_data_extract({extract_obj_name})\n"
             code += "\n"
 
         if include_imports:
@@ -649,7 +649,7 @@ class DataExtractor:
         code = "\n".join(filtered_lines)
 
         code += "\n"
-        code += "extract_output = extractor.create_extracts(name='extract data', description='lonerg description')\n"
+        code += f"extract_output = {obj_name}.create_extracts(name='extract data', description='longer description')\n"
         code += "status = extract_output.wait()\n"
         code += "if status == ExtractStatusType.COMPLETED:\n"
         code += "    extract_output.save_files('data_extracts_prefix', True)\n"