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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ee-client
-Version: 2.6.0
+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.0 → 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.0
+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.0 → 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.0 → ee_client-2.6.2}/eeclient/__init__.py

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

{ee_client-2.6.0 → ee_client-2.6.2}/eeclient/export/image.py

@@ -185,7 +185,13 @@ async def _export_image(
     image, updated_image_params, dimensions_consumed = image._apply_crs_and_affine(
         image_params
     )
-    image._apply_selection_and_scale(updated_image_params, dimensions_consumed)
+    # _apply_selection_and_scale returns a new image wrapped in
+    # clipToBoundsAndScale(geometry=region); discarding the return value would
+    # serialize the original unbounded image and cause GEE to reject the task
+    # with "Unable to export unbounded image." (issue #20)
+    image, _ = image._apply_selection_and_scale(
+        updated_image_params, dimensions_consumed
+    )
 
     expression = serializer.encode(image, for_cloud_api=True)
     request_params["expression"] = expression

{ee_client-2.6.0 → 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.0 → ee_client-2.6.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ee-client"
-version = "2.6.0"
+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.0"
+version = "2.6.2"
 version_files = [
     "pyproject.toml:version",
     "eeclient/__init__.py:__version__",

ee_client-2.6.2/tests/test_export_image.py

@@ -0,0 +1,89 @@
+from unittest.mock import AsyncMock, MagicMock, patch
+
+import pytest
+
+from eeclient.export.image import (
+    DriveDestination,
+    DriveOptions,
+    ImageFileFormat,
+    _export_image,
+)
+
+
+def test_drive_options_accepts_legacy_geotiff_alias():
+    opts = DriveOptions(
+        file_format="GeoTIFF",
+        drive_destination=DriveDestination(filename_prefix="out"),
+    )
+    assert opts.file_format == ImageFileFormat.GEO_TIFF
+
+
+def test_drive_options_accepts_canonical_value():
+    opts = DriveOptions(
+        file_format="GEO_TIFF",
+        drive_destination=DriveDestination(filename_prefix="out"),
+    )
+    assert opts.file_format == ImageFileFormat.GEO_TIFF
+
+
+def test_drive_options_rejects_unknown_string():
+    with pytest.raises(ValueError):
+        DriveOptions(
+            file_format="NotAFormat",
+            drive_destination=DriveDestination(filename_prefix="out"),
+        )
+
+
+@pytest.mark.asyncio
+async def test_export_image_uses_clipped_image_from_selection_and_scale():
+    """Regression test for issue #20.
+
+    `ee.Image._apply_selection_and_scale` returns a new image wrapped in
+    `clipToBoundsAndScale(geometry=region)`. The export flow must serialize
+    that clipped image, not the original, or GEE rejects the task as
+    "Unable to export unbounded image".
+    """
+    original_image = MagicMock(name="original_image")
+    crs_applied_image = MagicMock(name="crs_applied_image")
+    clipped_image = MagicMock(name="clipped_image")
+
+    original_image._apply_crs_and_affine.return_value = (
+        crs_applied_image,
+        {"region": "geom", "scale": 30},
+        False,
+    )
+    crs_applied_image._apply_selection_and_scale.return_value = (
+        clipped_image,
+        {},
+    )
+
+    client = MagicMock()
+    client.rest_call = AsyncMock(
+        return_value={"name": "projects/p/operations/op", "metadata": {}}
+    )
+
+    drive_options = DriveOptions(
+        file_format=ImageFileFormat.GEO_TIFF,
+        drive_destination=DriveDestination(filename_prefix="out"),
+    )
+
+    with (
+        patch("eeclient.export.image.serializer.encode") as mock_encode,
+        patch("eeclient.export.image.Task.model_validate") as mock_validate,
+    ):
+        mock_encode.return_value = {"result": "encoded"}
+        mock_validate.return_value = MagicMock()
+
+        await _export_image(
+            client=client,
+            image=original_image,
+            drive_options=drive_options,
+            region="geom",
+            scale=30,
+        )
+
+    # The serialized expression must be the clipped image, not the original
+    # or the intermediate crs-applied image.
+    mock_encode.assert_called_once()
+    encoded_image = mock_encode.call_args.args[0]
+    assert encoded_image is clipped_image

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

ee_client-2.6.0/tests/test_export_image.py

@@ -1,31 +0,0 @@
-import pytest
-
-from eeclient.export.image import (
-    DriveDestination,
-    DriveOptions,
-    ImageFileFormat,
-)
-
-
-def test_drive_options_accepts_legacy_geotiff_alias():
-    opts = DriveOptions(
-        file_format="GeoTIFF",
-        drive_destination=DriveDestination(filename_prefix="out"),
-    )
-    assert opts.file_format == ImageFileFormat.GEO_TIFF
-
-
-def test_drive_options_accepts_canonical_value():
-    opts = DriveOptions(
-        file_format="GEO_TIFF",
-        drive_destination=DriveDestination(filename_prefix="out"),
-    )
-    assert opts.file_format == ImageFileFormat.GEO_TIFF
-
-
-def test_drive_options_rejects_unknown_string():
-    with pytest.raises(ValueError):
-        DriveOptions(
-            file_format="NotAFormat",
-            drive_destination=DriveDestination(filename_prefix="out"),
-        )