luminarycloud 0.12.0__py3-none-any.whl → 0.13.1__py3-none-any.whl

luminarycloud/_client/client.py

@@ -12,6 +12,12 @@ from .._auth import Auth0Client
 from .._proto.api.v0.luminarycloud.geometry.geometry_pb2_grpc import GeometryServiceStub
 from .._proto.api.v0.luminarycloud.mesh.mesh_pb2_grpc import MeshServiceStub
 from .._proto.api.v0.luminarycloud.output_node.output_node_pb2_grpc import OutputNodeServiceStub
+from .._proto.api.v0.luminarycloud.output_definition.output_definition_pb2_grpc import (
+    OutputDefinitionServiceStub,
+)
+from .._proto.api.v0.luminarycloud.stopping_condition.stopping_condition_pb2_grpc import (
+    StoppingConditionServiceStub,
+)
 from .._proto.api.v0.luminarycloud.project.project_pb2_grpc import ProjectServiceStub
 from .._proto.api.v0.luminarycloud.simulation.simulation_pb2_grpc import (
     SimulationServiceStub,
@@ -42,6 +48,8 @@ class Client(
     UploadServiceStub,
     VisAPIServiceStub,
     OutputNodeServiceStub,
+    OutputDefinitionServiceStub,
+    StoppingConditionServiceStub,
 ):
     """
     Creates a Luminary API client.
@@ -183,7 +191,8 @@ class Client(
         SimulationTemplateServiceStub.__init__(self, self._channel)
         VisAPIServiceStub.__init__(self, self._channel)
         OutputNodeServiceStub.__init__(self, self._channel)
-
+        OutputDefinitionServiceStub.__init__(self, self._channel)
+        StoppingConditionServiceStub.__init__(self, self._channel)
         for name, value in self.__dict__.items():
             if isinstance(value, grpc.UnaryUnaryMultiCallable):
                 setattr(self, name, rpc_error(value))

luminarycloud/_client/retry_interceptor.py

@@ -52,6 +52,6 @@ class RetryInterceptor(UnaryUnaryClientInterceptor):
             call.result()
         except grpc.RpcError as e:
             if e.code() == grpc.StatusCode.UNAUTHENTICATED:
-                raise AuthenticationError() from None  # type: ignore[type-arg]
+                raise AuthenticationError(e.details(), e.code()) from None
             raise
         return call

luminarycloud/_client/rpc_error.py

@@ -27,7 +27,7 @@ def rpc_error(func: Callable[..., Any]) -> Callable[..., Any]:
             for code, exception in _CODE_TO_EXCEPTION.items():
                 if e.code() == code:
                     logger.debug(f"RPC failed: {e.code().name}", exc_info=True)
-                    raise exception(e.details()) from None
+                    raise exception(e.details(), code) from None
             raise e
 
     return wrapper

luminarycloud/_helpers/__init__.py

@@ -1,39 +1,39 @@
 # Copyright 2023-2024 Luminary Cloud, Inc. All Rights Reserved.
-from .create_geometry import (
-    create_geometry,
+from ._create_geometry import (
+    create_geometry as create_geometry,
 )
-from .create_simulation import (
-    create_simulation,
+from ._create_simulation import (
+    create_simulation as create_simulation,
 )
 from .download import (
-    download_surface_solution,
-    download_volume_solution,
-    download_surface_deformation_template,
-    download_surface_sensitivity_data,
-    save_file,
+    download_surface_solution as download_surface_solution,
+    download_volume_solution as download_volume_solution,
+    download_surface_deformation_template as download_surface_deformation_template,
+    download_surface_sensitivity_data as download_surface_sensitivity_data,
+    save_file as save_file,
 )
 from .file_chunk_stream import (
-    FileChunkStream,
+    FileChunkStream as FileChunkStream,
 )
-from .simulation_params_from_json import (
-    simulation_params_from_json,
-    simulation_params_from_json_path,
+from ._simulation_params_from_json import (
+    simulation_params_from_json as simulation_params_from_json,
+    simulation_params_from_json_path as simulation_params_from_json_path,
 )
-from .timestamp_to_datetime import (
-    timestamp_to_datetime,
+from ._timestamp_to_datetime import (
+    timestamp_to_datetime as timestamp_to_datetime,
 )
 from .upload import (
-    upload_file,
+    upload_file as upload_file,
 )
 from ._upload_mesh import (
-    upload_mesh,
-    upload_mesh_from_local_file,
-    upload_mesh_from_url,
+    upload_mesh as upload_mesh,
+    upload_mesh_from_local_file as upload_mesh_from_local_file,
+    upload_mesh_from_url as upload_mesh_from_url,
 )
 from ._upload_table import (
-    upload_table_as_json,
-    upload_c81_as_json,
+    upload_table_as_json as upload_table_as_json,
+    upload_c81_as_json as upload_c81_as_json,
 )
-from .wait_for_simulation import (
-    wait_for_simulation,
+from ._wait_for_simulation import (
+    wait_for_simulation as wait_for_simulation,
 )

luminarycloud/_helpers/{create_geometry.py → _create_geometry.py}

@@ -4,7 +4,7 @@ from luminarycloud._proto.upload import upload_pb2 as uploadpb
 from os import PathLike
 from .._client import Client
 from .upload import upload_file
-from typing import Any, Optional
+from typing import Optional
 from luminarycloud._helpers import util
 import uuid
 import random
@@ -17,18 +17,19 @@ logger = logging.getLogger(__name__)
 
 def create_geometry(
     client: Client,
-    cad_file_path: PathLike,
+    cad_file_path: PathLike | str,
     project_id: str,
     *,
     name: Optional[str] = None,
     scaling: Optional[float] = None,
     wait: bool = False,
+    convert_to_discrete: bool = False,
 ) -> geometrypb.Geometry:
 
     cad_file_meta = util.get_file_metadata(cad_file_path)
     logger.info(
         f"creating geometry in {project_id} by uploading file: {cad_file_meta.name}.{cad_file_meta.ext}, "
-        + f"size: {cad_file_meta.size} bytes, sha256: {cad_file_meta.sha256_checksum}, "
+        + f"size: {cad_file_meta.size} bytes, sha256: {str(cad_file_meta.sha256_checksum)}, "
         + f"crc32c: {cad_file_meta.crc32c_checksum}"
     )
 
@@ -55,6 +56,7 @@ def create_geometry(
             scaling=scaling,
             wait=False,
             request_id=str(uuid.uuid4()),
+            force_discrete=convert_to_discrete,
         )
     )
     geo = create_geo_res.geometry

luminarycloud/_helpers/{create_simulation.py → _create_simulation.py}

@@ -36,8 +36,8 @@ def create_simulation(
             description=description,
             simulation_options=SimulationOptions(
                 batch_processing=batch_processing,
-                gpu_type=gpu_type,
-                gpu_count=gpu_count,
+                gpu_type=gpu_type.value if gpu_type is not None else GPUType.UNSPECIFIED.value,
+                gpu_count=gpu_count or 0,
             ),
         )
     )

luminarycloud/_helpers/_get_project_id.py

@@ -0,0 +1,29 @@
+from ..simulation import get_simulation
+from ..geometry import Geometry
+from ..solution import Solution
+from ..mesh import Mesh
+
+
+def _get_project_id(entity: Geometry | Solution | Mesh) -> str | None:
+    """
+    A helper function to get the project id from various entities.
+    Returns the project id or None if the entity type is unknown.
+
+    Parameters
+    ----------
+    entiry: Geometry | Solution | Mesh
+        The entity to get the project id from.
+    """
+    if isinstance(entity, Solution):
+        solution: Solution = entity
+        sim = get_simulation(solution.simulation_id)
+        return sim.project_id
+    elif isinstance(entity, Mesh):
+        mesh: Mesh = entity
+        return mesh.project_id
+    elif isinstance(entity, Geometry):
+        geometry: Geometry = entity
+        project = geometry.project()
+        return project.id
+    else:
+        return None

luminarycloud/_helpers/{simulation_params_from_json.py → _simulation_params_from_json.py}

@@ -11,7 +11,7 @@ def simulation_params_from_json(text: Union[str, bytes]) -> SimulationParam:
     return Parse(text, SimulationParam(), ignore_unknown_fields=True)
 
 
-def simulation_params_from_json_path(json_path: PathLike) -> SimulationParam:
+def simulation_params_from_json_path(json_path: PathLike | str) -> SimulationParam:
     with open(json_path, "rb") as fp:
         json_bytes = fp.read()
     return simulation_params_from_json(json_bytes)

luminarycloud/_helpers/_upload_mesh.py

@@ -55,7 +55,7 @@ def _upload_mesh_from_path(
     project_id: str,
     filepath: Union[os.PathLike, str],
     size: int,
-    sha256_checksum: str,
+    sha256_checksum: bytes,
     crc32c_checksum: bytes,
     mesh_type: MeshType,
     *,
@@ -79,9 +79,9 @@ def _upload_mesh_from_path(
         appropriate filename extension.
     size: int
         The size of the stream to be read
-    sha256checksum: str
+    sha256checksum: bytes
         The SHA256 digest of the mesh file to upload
-    crc32c_checksum: str
+    crc32c_checksum: bytes
         The CRC32C digest of the mesh file to upload
     mesh_type : MeshType
         The file format of the mesh file.
@@ -110,7 +110,7 @@ def _upload_mesh_from_path(
         ext=file_ext,  # note: this must be the extension without the dot
         size=size,
         sha256_checksum=sha256_checksum,
-        crc32c_checksum=crc32c_checksum,
+        crc32c_checksum=crc32c_checksum.decode(),
     )
     create_upload_res: uploadpb.CreateUploadReply = client.CreateUpload(
         uploadpb.CreateUploadRequest(
@@ -118,10 +118,10 @@ def _upload_mesh_from_path(
             file_meta=file_metadata,
             resource_params=uploadpb.ResourceParams(
                 mesh_params=uploadpb.MeshParams(
-                    mesh_name=name,
+                    mesh_name=name or file_name_without_ext,
                     scaling=scaling if scaling is not None else 1.0,
-                    mesh_type=mesh_type,
-                    do_not_read_zones_openfoam=do_not_read_zones_openfoam,
+                    mesh_type=mesh_type.value,
+                    do_not_read_zones_openfoam=do_not_read_zones_openfoam or False,
                     # We force disconnect to True here because we are not exposing this parameter.
                     disconnect=True,
                 )

luminarycloud/_helpers/_upload_table.py

@@ -63,7 +63,7 @@ def upload_table_as_json(
 def upload_c81_as_json(
     client: Client,
     project_id: str,
-    c81: PathLike,
+    c81: PathLike | str,
 ) -> str:
     """
     Uploads a c81 table and converts it to RectilinearTable json.
@@ -74,7 +74,7 @@ def upload_c81_as_json(
         A LuminaryCloud Client (see client.py)
     project_id: str
         The ID of the project to upload the table under
-    c81: PathLike
+    c81: PathLike or str
         Path to c81 file to upload
 
     Returns

luminarycloud/_helpers/{wait_for_simulation.py → _wait_for_simulation.py}

@@ -75,7 +75,7 @@ def _print_residuals(client: Client, simulation_id: str, last_printed_iter: int)
     residuals: GetSimulationGlobalResidualsResponse = client.GetSimulationGlobalResiduals(
         GetSimulationGlobalResidualsRequest(id=simulation_id)
     )
-    with io.StringIO(residuals.csv_file.data.decode()) as stream:
+    with io.StringIO(residuals.csv_file.full_contents.decode()) as stream:
         if last_printed_iter < 0:  # we haven't printed the header yet
             first_line = stream.readline().strip().split(",")
             header = "\t".join("{:>18}".format(s) for s in first_line)