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

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

@@ -56,14 +56,14 @@ class PhysicsAiArchitectureVersion(ProtoWrapperBase):
             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 "process_cpus" not in config["resources"]:
+            config["resources"]["process_cpus"] = 8
         if "train_gpus" not in config["resources"]:
-            config["resources"]["train_gpus"] = 8
+            config["resources"]["train_gpus"] = 1
         if "test_gpus" not in config["resources"]:
-            config["resources"]["test_gpus"] = 8
+            config["resources"]["test_gpus"] = 1
         if "mode" not in config:
-            config["mode"] = "full-gpu"
+            config["mode"] = "full"
 
         training_config_json = json.dumps(config, indent=2)
         external_dataset_uri = f"gs://training-data/architecture-{self.id}"

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

@@ -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,
 )

luminarycloud/pipelines/api.py

@@ -1,15 +1,35 @@
 # Copyright 2023-2024 Luminary Cloud, Inc. All Rights Reserved.
-from typing import Literal
+from typing import Any, Literal
 from dataclasses import dataclass
 
 from datetime import datetime
 
+from .arguments import PipelineArgValueType
 from ..pipelines import Pipeline, PipelineArgs
 from .._client import get_default_client
 
 
+@dataclass
+class LogLine:
+    timestamp: datetime
+    level: int
+    message: str
+
+    @classmethod
+    def from_json(cls, json: dict) -> "LogLine":
+        return cls(
+            timestamp=datetime.fromisoformat(json["timestamp"]),
+            level=json["level"],
+            message=json["message"],
+        )
+
+
 @dataclass
 class PipelineRecord:
+    """
+    A PipelineRecord represents a persisted pipeline.
+    """
+
     id: str
     name: str
     description: str | None
@@ -17,8 +37,10 @@ class PipelineRecord:
     created_at: datetime
     updated_at: datetime
 
-    def pipeline(self) -> Pipeline:
-        return Pipeline._from_yaml(self.definition_yaml)
+    # I don't think users need to get the Pipeline object from a PipelineRecord, but if they did,
+    # it would be done like this.
+    # def pipeline(self) -> Pipeline:
+    #     return Pipeline._from_yaml(self.definition_yaml)
 
     @classmethod
     def from_json(cls, json: dict) -> "PipelineRecord":
@@ -31,14 +53,30 @@ class PipelineRecord:
             updated_at=datetime.fromisoformat(json["updated_at"]),
         )
 
+    def pipeline_jobs(self) -> "list[PipelineJobRecord]":
+        """
+        Returns a list of pipeline jobs that were created from this pipeline.
+
+        Returns
+        -------
+        list[PipelineJobRecord]
+            A list of PipelineJobRecord objects.
+        """
+        res = get_default_client().http.get(f"/rest/v0/pipelines/{self.id}/pipeline_jobs")
+        return [PipelineJobRecord.from_json(p) for p in res["data"]]
+
 
 @dataclass
 class PipelineJobRecord:
+    """
+    A PipelineJobRecord represents a persisted pipeline job.
+    """
+
     id: str
     pipeline_id: str
     name: str
     description: str | None
-    status: Literal["pending", "running", "completed", "failed", "cancelled"]
+    status: Literal["pending", "running", "completed", "failed"]
     created_at: datetime
     updated_at: datetime
     started_at: datetime | None
@@ -60,6 +98,123 @@ class PipelineJobRecord:
             ),
         )
 
+    def pipeline(self) -> PipelineRecord:
+        """
+        Returns the pipeline that this pipeline job was created from.
+
+        Returns
+        -------
+        PipelineRecord
+            The PipelineRecord for the pipeline that this pipeline job was created from.
+        """
+        return get_pipeline(self.pipeline_id)
+
+    def runs(self) -> "list[PipelineJobRunRecord]":
+        """
+        Returns a list of runs for this pipeline job.
+
+        Returns
+        -------
+        list[PipelineJobRunRecord]
+            A list of PipelineJobRunRecord objects.
+        """
+        res = get_default_client().http.get(f"/rest/v0/pipeline_jobs/{self.id}/runs")
+        return [PipelineJobRunRecord.from_json(r) for r in res["data"]]
+
+    def logs(self) -> list[LogLine]:
+        """
+        Returns a list of log lines for this pipeline job.
+
+        Each log line is a LogLine object, which has a timestamp, level, and message.
+
+        Returns
+        -------
+        list[LogLine]
+            A list of LogLine objects.
+        """
+        res = get_default_client().http.get(f"/rest/v0/pipeline_jobs/{self.id}/logs")
+        return [LogLine.from_json(l) for l in res["data"]]
+
+    def artifacts(self) -> list[dict]:
+        """
+        Returns a list of artifacts that were produced by this pipeline job.
+
+        Artifacts are things like Geometries, Meshes, and Simulations. Each artifact is a dictionary
+        with an "id" key, which is an identifier for the artifact.
+
+        .. warning:: This feature is experimental and may change or be removed in the future.
+
+        Returns
+        -------
+        list[dict]
+            A list of artifact dictionaries.
+        """
+        res = get_default_client().http.get(f"/rest/v0/pipeline_jobs/{self.id}/artifacts")
+        return res["data"]
+
+
+@dataclass
+class PipelineJobRunRecord:
+    pipeline_job_id: str
+    idx: int
+    arguments: list[PipelineArgValueType]
+    status: Literal["pending", "running", "completed", "failed"]
+
+    @classmethod
+    def from_json(cls, json: dict) -> "PipelineJobRunRecord":
+        return cls(
+            pipeline_job_id=json["pipeline_job_id"],
+            idx=json["idx"],
+            arguments=json["arguments"],
+            status=json["status"],
+        )
+
+    def pipeline_job(self) -> PipelineJobRecord:
+        """
+        Returns the pipeline job that this pipeline job run was created from.
+
+        Returns
+        -------
+        PipelineJobRecord
+            The PipelineJobRecord for the pipeline job that this pipeline job run was created from.
+        """
+        return get_pipeline_job(self.pipeline_job_id)
+
+    def logs(self) -> list[LogLine]:
+        """
+        Returns a list of log lines for this pipeline job run.
+
+        Each log line is a LogLine object, which has a timestamp, level, and message.
+
+        Returns
+        -------
+        list[LogLine]
+            A list of LogLine objects.
+        """
+        res = get_default_client().http.get(
+            f"/rest/v0/pipeline_jobs/{self.pipeline_job_id}/runs/{self.idx}/logs"
+        )
+        return [LogLine.from_json(l) for l in res["data"]]
+
+    def artifacts(self) -> list[dict]:
+        """
+        Returns a list of artifacts that were produced by this pipeline job run.
+
+        Artifacts are things like Geometries, Meshes, and Simulations. Each artifact is a dictionary
+        with an "id" key, which is an identifier for the artifact.
+
+        .. warning:: This feature is experimental and may change or be removed in the future.
+
+        Returns
+        -------
+        list[dict]
+            A list of artifact dictionaries.
+        """
+        res = get_default_client().http.get(
+            f"/rest/v0/pipeline_jobs/{self.pipeline_job_id}/runs/{self.idx}/artifacts"
+        )
+        return res["data"]
+
 
 def create_pipeline(
     name: str, pipeline: Pipeline | str, description: str | None = None

luminarycloud/pipelines/arguments.py

@@ -103,3 +103,18 @@ class PipelineArgs:
         return (
             f"PipelineArgs(param_names={[p.name for p in self.params]}, row_count={len(self.rows)})"
         )
+
+    def print_as_table(self) -> None:
+        headers = [p.name for p in self.params]
+        row_strs = [[str(v) for v in row.row_values] for row in self.rows]
+        col_widths = [
+            max(len(headers[i]), *(len(r[i]) for r in row_strs)) for i in range(len(headers))
+        ]
+
+        def format_row(values: list[str]) -> str:
+            return " | ".join(val.ljust(col_widths[i]) for i, val in enumerate(values))
+
+        print(format_row(headers))
+        print("-+-".join("-" * w for w in col_widths))
+        for r in row_strs:
+            print(format_row(r))

luminarycloud/pipelines/operators.py

@@ -2,7 +2,7 @@
 from dataclasses import dataclass
 
 from .core import Operator, OperatorInputs, OperatorOutputs, PipelineOutput
-from .parameters import StringPipelineParameter
+from .parameters import StringPipelineParameter, IntPipelineParameter
 from ..meshing import MeshGenerationParams
 
 
@@ -43,6 +43,8 @@ class ReadGeometry(Operator[ReadGeometryOutputs]):
     """
     Reads a Geometry into the Pipeline.
 
+    .. warning:: This feature is experimental and may change or be removed in the future.
+
     Parameters
     ----------
     geometry_id : str | StringPipelineParameter
@@ -52,8 +54,6 @@ class ReadGeometry(Operator[ReadGeometryOutputs]):
     -------
     geometry : PipelineOutputGeometry
         The latest GeometryVersion of the Geometry as of the moment the Pipeline was invoked.
-
-    .. warning:: This feature is experimental and may change or be removed in the future.
     """
 
     def __init__(
@@ -70,6 +70,51 @@ class ReadGeometry(Operator[ReadGeometryOutputs]):
         )
 
 
+@dataclass
+class ReadMeshOutputs(OperatorOutputs):
+    mesh: PipelineOutputMesh
+    """
+    The Mesh read from the given `mesh_id`.
+    """
+
+
+class ReadMesh(Operator[ReadMeshOutputs]):
+    """
+    Reads a Mesh into the Pipeline.
+
+    .. warning:: This feature is experimental and may change or be removed in the future.
+
+    Parameters
+    ----------
+    mesh_id : str | StringPipelineParameter
+        The ID of the Mesh to retrieve.
+    wait_timeout_seconds : int | IntPipelineParameter | None
+        The number of seconds to wait for the Mesh to be ready. If None, defaults to 1800 seconds
+        (30 minutes).
+
+    Outputs
+    -------
+    mesh : PipelineOutputMesh
+        The Mesh with the given `mesh_id`.
+    """
+
+    def __init__(
+        self,
+        *,
+        task_name: str | None = None,
+        mesh_id: str | StringPipelineParameter,
+        wait_timeout_seconds: int | IntPipelineParameter | None = None,
+    ):
+        if wait_timeout_seconds is None:
+            wait_timeout_seconds = 30 * 60
+        super().__init__(
+            task_name,
+            {"mesh_id": mesh_id, "wait_timeout_seconds": wait_timeout_seconds},
+            OperatorInputs(self),
+            ReadMeshOutputs._instantiate_for(self),
+        )
+
+
 @dataclass
 class ModifyGeometryOutputs(OperatorOutputs):
     geometry: PipelineOutputGeometry
@@ -81,6 +126,8 @@ class ModifyGeometry(Operator[ModifyGeometryOutputs]):
     """
     Modifies a Geometry.
 
+    .. warning:: This feature is experimental and may change or be removed in the future.
+
     Parameters
     ----------
     mods : dict
@@ -92,8 +139,6 @@ class ModifyGeometry(Operator[ModifyGeometryOutputs]):
     -------
     geometry : PipelineOutputGeometry
         The modified Geometry, represented as a new GeometryVersion.
-
-    .. warning:: This feature is experimental and may change or be removed in the future.
     """
 
     def __init__(
@@ -122,31 +167,37 @@ class Mesh(Operator[MeshOutputs]):
     """
     Generates a Mesh from a Geometry.
 
+    .. warning:: This feature is experimental and may change or be removed in the future.
+
     Parameters
     ----------
-    target_cv_count : int | None
-        The target number of control volumes to generate. If None, a minimal mesh will be generated.
     geometry : PipelineOutputGeometry
         The Geometry to mesh.
+    mesh_name : str | StringPipelineParameter | None
+        The name to assign to the Mesh. If None, a default name will be used.
+    target_cv_count : int | None
+        The target number of control volumes to generate. If None, a minimal mesh will be generated.
 
     Outputs
     -------
     mesh : PipelineOutputMesh
         The generated Mesh.
-
-    .. warning:: This feature is experimental and may change or be removed in the future.
     """
 
     def __init__(
         self,
         *,
         task_name: str | None = None,
-        target_cv_count: int | None,
         geometry: PipelineOutputGeometry,
+        mesh_name: str | StringPipelineParameter | None = None,
+        target_cv_count: int | None,
     ):
         super().__init__(
             task_name,
-            {"target_cv_count": target_cv_count},
+            {
+                "mesh_name": mesh_name,
+                "target_cv_count": target_cv_count,
+            },
             OperatorInputs(self, geometry=(PipelineOutputGeometry, geometry)),
             MeshOutputs._instantiate_for(self),
         )
@@ -169,31 +220,37 @@ class Simulate(Operator[SimulateOutputs]):
     """
     Runs a Simulation.
 
+    .. warning:: This feature is experimental and may change or be removed in the future.
+
     Parameters
     ----------
-    sim_template_id : str | StringPipelineParameter
-        The ID of the SimulationTemplate to use for the Simulation.
     mesh : PipelineOutputMesh
         The Mesh to use for the Simulation.
+    sim_name : str | StringPipelineParameter | None
+        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.
 
     Outputs
     -------
     simulation : PipelineOutputSimulation
         The Simulation.
-
-    .. warning:: This feature is experimental and may change or be removed in the future.
     """
 
     def __init__(
         self,
         *,
         task_name: str | None = None,
-        sim_template_id: str | StringPipelineParameter,
         mesh: PipelineOutputMesh,
+        sim_name: str | StringPipelineParameter | None = None,
+        sim_template_id: str | StringPipelineParameter,
     ):
         super().__init__(
             task_name,
-            {"sim_template_id": sim_template_id},
+            {
+                "sim_name": sim_name,
+                "sim_template_id": sim_template_id,
+            },
             OperatorInputs(self, mesh=(PipelineOutputMesh, mesh)),
             SimulateOutputs._instantiate_for(self),
         )

luminarycloud/project.py

@@ -12,6 +12,7 @@ import concurrent
 
 import luminarycloud as lc
 from luminarycloud._helpers.named_variables import _named_variables_to_proto
+from luminarycloud._helpers.pagination import PaginationIterator
 from luminarycloud.params.simulation.adjoint_ import Adjoint
 
 from ._client import get_default_client
@@ -785,53 +786,13 @@ def list_projects() -> list[Project]:
     return list(iterate_projects())
 
 
-class ProjectIterator:
+class ProjectIterator(PaginationIterator[Project]):
     """Iterator class for projects that provides length hint."""
 
-    def __init__(self, page_size: int):
-        self._page_size: int = page_size
-        self._page_token: str = ""
-        self._total_count: Optional[int] = None
-        self._current_page: Optional[list[projectpb.Project]] = None
-        self._client = get_default_client()
-        self._iterated_count: int = 0
-
-    def __iter__(self) -> "ProjectIterator":
-        return self
-
-    def __next__(self) -> Project:
-        if self._current_page is None:
-            self._fetch_next_page()
-
-        # _current_page really can't be None here, but this assertion is needed to appease mypy
-        assert self._current_page is not None
-
-        if len(self._current_page) == 0:
-            if not self._page_token:
-                raise StopIteration
-            self._fetch_next_page()
-
-        self._iterated_count += 1
-
-        return Project(self._current_page.pop(0))
-
-    def _fetch_next_page(self) -> None:
-        req = projectpb.ListProjectsRequest(page_size=self._page_size, page_token=self._page_token)
+    def _fetch_page(self, page_size: int, page_token: str) -> tuple[list[Project], str, int]:
+        req = projectpb.ListProjectsRequest(page_size=page_size, page_token=page_token)
         res = self._client.ListProjects(req)
-
-        self._current_page = list(res.projects)
-        self._page_token = res.next_page_token
-
-        # Set length hint on first fetch if available
-        if self._total_count is None:
-            self._total_count = res.total_count or 0
-
-    def __length_hint__(self) -> int:
-        if self._total_count is None:
-            # Fetch first page to get total size if not already fetched
-            if self._current_page is None:
-                self._fetch_next_page()
-        return max(0, (self._total_count or 0) - self._iterated_count)
+        return [Project(p) for p in res.projects], res.next_page_token, res.total_count
 
 
 def iterate_projects(page_size: int = 50) -> ProjectIterator: