luminarycloud 0.15.3__py3-none-any.whl → 0.15.4__py3-none-any.whl

luminarycloud/params/simulation/physics/heat_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr
@@ -155,7 +154,8 @@ class Heat(CodeRepr, ParamGroupWrapper[clientpb.Heat]):
             self.initialization = HeatPrescribedValues()
         elif proto.initialization_heat.initialization_type == clientpb.EXISTING_SOLUTION:
             self.initialization = HeatExistingSolution()
-        self.initialization._from_proto(proto.initialization_heat)
+        if self.initialization is not None:
+            self.initialization._from_proto(proto.initialization_heat)
         self.heat_source = [HeatSource.from_proto(v) for v in proto.heat_source]
         self.sliding_interfaces = [
             SlidingInterfaces.from_proto(v) for v in proto.sliding_interfaces

luminarycloud/params/simulation/physics/periodic_pair/periodicity_type/rotational_periodicity_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/physics/periodic_pair/periodicity_type/translational_periodicity_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/physics/periodic_pair/periodicity_type_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/physics/periodic_pair_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/physics/solution_controls/linear_solver_type/gauss_seidel_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/physics/solution_controls/linear_solver_type/krylov_amg_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr
@@ -36,7 +35,7 @@ class KrylovAmg(LinearSolverType):
     "Linear Solver AMG Coarsening Sizing."
     levels: int = 20
     "Linear Solver AMG Level."
-    relaxation: float = 0.75
+    relaxation: LcFloat = 0.75
     "Linear Solver AMG Relaxation applied to the smoother."
     multigrid_cycle: enum.LinsolAmgCycleType = enum.LinsolAmgCycleType.LINSOL_AMG_CYCLE_TYPE_V
     "AMG cycle type."

luminarycloud/params/simulation/physics/solution_controls/linear_solver_type_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr
@@ -25,5 +24,5 @@ class LinearSolverType(CodeRepr, metaclass=ABCMeta):
 
     iterations: int = 15
     "Maximum number of linear solver iterations per nonlinear solution update. The linear solver terminates if the maximum is reached."
-    tolerance: float = 0.05
+    tolerance: LcFloat = 0.05
     "Relative reduction target in the linear solver residual from an initial value of 1.0 per nonlinear solution update. The linear solver terminates if the tolerance is met."

luminarycloud/params/simulation/physics_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr
@@ -28,7 +27,9 @@ from luminarycloud.params.simulation.physics.heat_ import *
 class Physics(CodeRepr, ParamGroupWrapper[clientpb.Physics]):
     """Single physics solver entity."""
 
-    physics_identifier: EntityIdentifier | None = None
+    physics_identifier: EntityIdentifier = field(
+        default_factory=_create_entity_identifier, init=True
+    )
     "Unique identifier for a physics entity."
     heat: Heat | None = None
     "Configuration for a heat transfer physics solver - simulates heat transfer in solid media."

luminarycloud/params/simulation/simulation_param_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/sliding_interfaces_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/surface_name_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/compute_statistics/compute_statistics_off_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/compute_statistics/compute_statistics_on_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/compute_statistics_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/time_marching/time_explicit_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/time_marching/time_implicit_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/time_marching_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/time_step_ramp/time_step_ramp_off_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time/time_step_ramp/time_step_ramp_on_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr
@@ -26,7 +25,7 @@ from luminarycloud.params.simulation.time.time_step_ramp_ import *
 class TimeStepRampOn(TimeStepRamp):
     """Enable physical time step ramping."""
 
-    initial_time_step: float = 0.001
+    initial_time_step: LcFloat = 0.001
     "Initial physical time step for time step ramping. The physical time step is ramped toward the target for the duration of the ramp."
     start_iteration: int = 500
     "Time step iteration to start the time step ramping process."

luminarycloud/params/simulation/time/time_step_ramp_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr

luminarycloud/params/simulation/time_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr
@@ -48,7 +47,7 @@ class Time(CodeRepr, ParamGroupWrapper[clientpb.Time]):
 
     time_marching: TimeMarching = field(default_factory=TimeImplicit)
     "Scheme for time-accurate integration. Possible types: ``TimeImplicit``, ``TimeExplicit`` from the ``time_marching`` module."
-    time_step: float = 0.0001
+    time_step: LcFloat = 0.0001
     "The fixed physical time step."
     time_step_ramp: TimeStepRamp = field(default_factory=TimeStepRampOff)
     "Use a larger time step value during the initial transients of a simulation and then ramp linearly towards the target value, to accelerate statistical convergence. Only applicable to transient problems with time implicit integration (dual time stepping). Possible types: ``TimeStepRampOff``, ``TimeStepRampOn`` from the ``time_step_ramp`` module."

luminarycloud/params/simulation/volume_entity_.py

@@ -5,12 +5,11 @@ from dataclasses import dataclass, field
 from typing import Any
 from uuid import uuid4
 
-from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
 from google.protobuf.message import Message as _Message
-
-from luminarycloud.tables import RectilinearTable
-from luminarycloud.types import Vector3
+from luminarycloud.tables import RectilinearTable, _param_name_to_table_type
+from luminarycloud.types import Vector3, LcFloat
 from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud._helpers._entity_identifier import _create_entity_identifier
 from luminarycloud._proto.client import simulation_pb2 as clientpb
 from luminarycloud._proto.client.entity_pb2 import EntityIdentifier
 from luminarycloud._helpers import CodeRepr
@@ -23,7 +22,9 @@ from luminarycloud.params.simulation._lib import ParamGroupWrapper, create_uniqu
 class VolumeEntity(CodeRepr, ParamGroupWrapper[clientpb.VolumeEntity]):
     """Volume entity."""
 
-    volume_identifier: EntityIdentifier | None = None
+    volume_identifier: EntityIdentifier = field(
+        default_factory=_create_entity_identifier, init=True
+    )
     "Unique identifier for a volume entity."
 
     def _to_proto(self) -> clientpb.VolumeEntity:

luminarycloud/physics_ai/__init__.py

@@ -1,8 +1,10 @@
 from .architectures import (
     list_architectures as list_architectures,
     PhysicsAiArchitecture as PhysicsAiArchitecture,
+    PhysicsAiArchitectureVersion as PhysicsAiArchitectureVersion,
 )
 from .models import (
     list_pretrained_models as list_pretrained_models,
     PhysicsAiModel as PhysicsAiModel,
+    PhysicsAiModelVersion as PhysicsAiModelVersion,
 )

luminarycloud/physics_ai/architectures.py

@@ -1,37 +1,71 @@
 # Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
-from __future__ import annotations
-
-from typing import List
+from typing import List, Optional
 
 from .._client import get_default_client
 from .._proto.api.v0.luminarycloud.physics_ai import physics_ai_pb2 as physaipb
 from .._wrapper import ProtoWrapper, ProtoWrapperBase
 from .._helpers.warnings import experimental
-from ..types.ids import PhysicsAiArchitectureID
+from ..types.ids import PhysicsAiArchitectureID, PhysicsAiArchitectureVersionID
+from ..enum.physics_ai_lifecycle_state import PhysicsAiLifecycleState
+
+
+@experimental
+@ProtoWrapper(physaipb.PhysicsAiArchitectureVersion)
+class PhysicsAiArchitectureVersion(ProtoWrapperBase):
+    """
+    Represents a specific version of a Physics AI architecture.
+
+    .. warning:: This feature is experimental and may change or be removed without notice.
+    """
+
+    id: PhysicsAiArchitectureVersionID
+    name: str
+    changelog: str
+    lifecycle_state: PhysicsAiLifecycleState
+    _proto: physaipb.PhysicsAiArchitectureVersion
 
 
 @experimental
 @ProtoWrapper(physaipb.PhysicsAiArchitecture)
 class PhysicsAiArchitecture(ProtoWrapperBase):
     """
-    Represents a Physics AI architecture.
+    Represents a Physics AI architecture with all its versions.
 
-    warning:: This feature is experimental and may change or be removed without notice.
+    .. warning:: This feature is experimental and may change or be removed without notice.
     """
 
     id: PhysicsAiArchitectureID
     name: str
     description: str
-    version: str
+    versions: List[PhysicsAiArchitectureVersion]
     _proto: physaipb.PhysicsAiArchitecture
 
+    def get_latest_version(self) -> Optional[PhysicsAiArchitectureVersion]:
+        """
+        Get the latest version of this architecture based on name.
+
+        Returns
+        -------
+        PhysicsAiArchitectureVersion or None
+            The first architecture version, or None if no versions exist.
+            Note: Version ordering is now determined by the backend.
+        """
+        if not self.versions:
+            return None
+        return self.versions[0] if self.versions else None
+
 
 @experimental
 def list_architectures() -> List[PhysicsAiArchitecture]:
     """
     List available Physics AI architectures for model training.
 
-    warning:: This feature is experimental and may change or be removed without notice.
+    .. warning:: This feature is experimental and may change or be removed without notice.
+
+    Returns
+    -------
+    list[PhysicsAiArchitecture]
+        A list of all available Physics AI architectures.
     """
     req = physaipb.ListArchitecturesRequest()
     res = get_default_client().ListArchitectures(req)

luminarycloud/physics_ai/inference.py

@@ -1,7 +1,5 @@
 # File: python/sdk/luminarycloud/inference/inference.py
 # Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
-from __future__ import annotations
-
 from datetime import datetime
 from typing import Any
 from json import loads as json_loads

luminarycloud/physics_ai/models.py

@@ -1,41 +1,70 @@
 # Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
-from __future__ import annotations
-
-from datetime import datetime
-from typing import List
+from typing import List, Optional
 
 from .._client import get_default_client
-from .._helpers._timestamp_to_datetime import timestamp_to_datetime
 from .._proto.api.v0.luminarycloud.physics_ai import physics_ai_pb2 as physaipb
 from .._wrapper import ProtoWrapper, ProtoWrapperBase
 from .._helpers.warnings import experimental
-from ..types.ids import PhysicsAiModelID
+from ..types.ids import PhysicsAiModelID, PhysicsAiModelVersionID
+from ..enum.physics_ai_lifecycle_state import PhysicsAiLifecycleState
+
+
+@experimental
+@ProtoWrapper(physaipb.PhysicsAiModelVersion)
+class PhysicsAiModelVersion(ProtoWrapperBase):
+    """
+    Represents a specific version of a Physics AI model.
+
+    .. warning:: This feature is experimental and may change or be removed without notice.
+    """
+
+    id: PhysicsAiModelVersionID
+    name: str
+    lifecycle_state: PhysicsAiLifecycleState
+    _proto: physaipb.PhysicsAiModelVersion
 
 
 @experimental
 @ProtoWrapper(physaipb.PhysicsAiModel)
 class PhysicsAiModel(ProtoWrapperBase):
-    """Represents a Physics AI model."""
+    """
+    Represents a Physics AI model with all its versions.
+
+    .. warning:: This feature is experimental and may change or be removed without notice.
+    """
 
     id: PhysicsAiModelID
     name: str
-    version: str
-
+    description: str
+    versions: List[PhysicsAiModelVersion]
     _proto: physaipb.PhysicsAiModel
 
-    @property
-    def create_time(self) -> datetime:
-        return timestamp_to_datetime(self._proto.create_time)
+    def get_latest_version(self) -> Optional[PhysicsAiModelVersion]:
+        """
+        Get the latest version of this model.
 
-    @property
-    def update_time(self) -> datetime:
-        return timestamp_to_datetime(self._proto.update_time)
+        Returns
+        -------
+        PhysicsAiModelVersion or None
+            The first model version, or None if no versions exist.
+            Note: Version ordering is now determined by the backend.
+        """
+        if not self.versions:
+            return None
+        return self.versions[0] if self.versions else None
 
 
 @experimental
 def list_pretrained_models() -> List[PhysicsAiModel]:
     """
     List available pretrained Physics AI models.
+
+    .. warning:: This feature is experimental and may change or be removed without notice.
+
+    Returns
+    -------
+    list[PhysicsAiModel]
+        A list of all available pretrained Physics AI models.
     """
     req = physaipb.ListPretrainedModelsRequest()
     res = get_default_client().ListPretrainedModels(req)