ee-client 2.6.1__tar.gz → 2.6.2__tar.gz

{ee_client-2.6.1 → ee_client-2.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ee-client
-Version: 2.6.1
+Version: 2.6.2
 Summary: extends the capabilities of the earthengine-api by providing custom session management and client interactions
 Author-email: Daniel Guerrero <dfgm2006@gmail.com>
 Project-URL: Homepage, https://github.com/dfguerrerom/ee-client

{ee_client-2.6.1 → ee_client-2.6.2}/ee_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ee-client
-Version: 2.6.1
+Version: 2.6.2
 Summary: extends the capabilities of the earthengine-api by providing custom session management and client interactions
 Author-email: Daniel Guerrero <dfgm2006@gmail.com>
 Project-URL: Homepage, https://github.com/dfguerrerom/ee-client

{ee_client-2.6.1 → ee_client-2.6.2}/ee_client.egg-info/SOURCES.txt

@@ -29,4 +29,5 @@ tests/test_data.py
 tests/test_export_image.py
 tests/test_export_table.py
 tests/test_integration_get_assets.py
-tests/test_models.py
+tests/test_models.py
+tests/test_tasks.py

{ee_client-2.6.1 → ee_client-2.6.2}/eeclient/__init__.py

@@ -1,6 +1,6 @@
 __title__ = "eeclient"
 __summary__ = "A client for Google Earth Engine"
-__version__ = "2.6.1"
+__version__ = "2.6.2"
 
 __author__ = "Daniel Guerrero"
 __email__ = "dfgm2006@gmail.com"

{ee_client-2.6.1 → ee_client-2.6.2}/eeclient/tasks.py

@@ -24,9 +24,12 @@ class CamelCaseModel(BaseModel):
 
 class Stage(CamelCaseModel):
     display_name: str
-    complete_work_units: int
-    total_work_units: str
-    description: str
+    # The Cloud API omits these fields on a stage until that stage has
+    # reported progress, so they must be optional to avoid breaking polling
+    # of a freshly-RUNNING task.
+    complete_work_units: Optional[int] = None
+    total_work_units: Optional[str] = None
+    description: Optional[str] = None
 
 
 class TaskMetadata(CamelCaseModel):

{ee_client-2.6.1 → ee_client-2.6.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ee-client"
-version = "2.6.1"
+version = "2.6.2"
 description = "extends the capabilities of the earthengine-api by providing custom session management and client interactions"
 readme = { file = "README.rst",  content-type = "text/x-rst" }
 authors = [
@@ -66,7 +66,7 @@ branch = true
 [tool.commitizen]
 tag_format = "v$major.$minor.$patch$prerelease"
 update_changelog_on_bump = false
-version = "2.6.1"
+version = "2.6.2"
 version_files = [
     "pyproject.toml:version",
     "eeclient/__init__.py:__version__",

ee_client-2.6.2/tests/test_tasks.py

@@ -0,0 +1,60 @@
+"""Tests for the Task / Stage models in eeclient.tasks."""
+
+from eeclient.tasks import Stage, Task
+
+
+def test_stage_allows_missing_work_units():
+    """The Cloud API omits completeWorkUnits/totalWorkUnits on a stage until
+    that stage reports progress; the model must accept this."""
+    stage = Stage.model_validate(
+        {
+            "displayName": "Create List of Assets",
+            "description": "Listing of temporary files.",
+        }
+    )
+    assert stage.display_name == "Create List of Assets"
+    assert stage.complete_work_units is None
+    assert stage.total_work_units is None
+
+
+def test_stage_allows_missing_description():
+    """Some stages omit `description` entirely before reporting progress."""
+    stage = Stage.model_validate({"displayName": "Write Files"})
+    assert stage.display_name == "Write Files"
+    assert stage.description is None
+    assert stage.complete_work_units is None
+    assert stage.total_work_units is None
+
+
+def test_task_validates_with_incomplete_stages():
+    """Regression for issue #22: polling a freshly-RUNNING export returns
+    stages without completeWorkUnits / totalWorkUnits and must not raise."""
+    payload = {
+        "name": "projects/ee-project/operations/ABC123",
+        "metadata": {
+            "@type": "type.googleapis.com/google.earthengine.v1.OperationMetadata",
+            "state": "RUNNING",
+            "description": "my-export",
+            "priority": 100,
+            "createTime": "2026-05-13T12:00:00Z",
+            "type": "EXPORT_IMAGE",
+            "stages": [
+                {
+                    "displayName": "Create List of Assets",
+                    "description": "Listing of temporary files.",
+                },
+                {
+                    "displayName": "Write Files",
+                    "description": "Writing files to the export destination.",
+                },
+            ],
+        },
+        "done": False,
+    }
+
+    task = Task.model_validate(payload)
+    assert task.id == "ABC123"
+    assert task.metadata.stages is not None
+    assert len(task.metadata.stages) == 2
+    assert task.metadata.stages[0].complete_work_units is None
+    assert task.metadata.stages[0].total_work_units is None