From 211c052a78919bd6804f6883a135c99d1759f725 Mon Sep 17 00:00:00 2001
From: Sergio Souza Costa <sergio.costa@ufma.br>
Date: Mon, 15 Jun 2026 11:51:02 -0300
Subject: [PATCH] fix: handle temporal state vars in synchronize(); silence NaN
 cast warning
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- sync_model.py: SyncRasterModel.synchronize() now takes arr[0] when a
  state variable is stored as (time, y, x) — happens when to_lucc_data()
  loads a variable that has temporal catalog entries. After the first step
  the model writes back 2D arrays so subsequent snapshots are 2D.
- io/raster.py: fill NaN/inf with the band nodata value before casting to
  integer dtype, eliminating the RuntimeWarning from float→int16 cast when
  raster arrays contain masked cells.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 dissmodel/geo/raster/sync_model.py | 10 +++++++++-
 dissmodel/io/raster.py             |  2 ++
 pyproject.toml                     |  2 +-
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/dissmodel/geo/raster/sync_model.py b/dissmodel/geo/raster/sync_model.py
index 967a87d..08587d4 100644
--- a/dissmodel/geo/raster/sync_model.py
+++ b/dissmodel/geo/raster/sync_model.py
@@ -120,8 +120,16 @@ def synchronize(self) -> None:
 
         Does nothing if ``land_use_types`` has not been set yet (safe to
         call before ``setup()`` completes).
+
+        If a state variable was loaded from a temporal catalog entry (shape
+        ``(time, y, x)``), the first slice is used as the initial state.
+        After the first step the model always writes back 2D arrays, so
+        subsequent snapshots are 2D unconditionally.
         """
         if not hasattr(self, "land_use_types"):
             return
         for name in self.land_use_types:
-            self.backend.set(name + "_past", self.backend.get(name).copy())
+            arr = self.backend.get(name)
+            if arr.ndim > 2:
+                arr = arr[0]
+            self.backend.set(name + "_past", arr.copy())
diff --git a/dissmodel/io/raster.py b/dissmodel/io/raster.py
index 07754d0..28d887e 100644
--- a/dissmodel/io/raster.py
+++ b/dissmodel/io/raster.py
@@ -209,6 +209,8 @@ def _write_geotiff(
                 name,
                 np.full((rows, cols), nodata, dtype=dtype),
             )
+            if np.issubdtype(np.dtype(dtype), np.integer):
+                arr = np.where(np.isfinite(arr), arr, nodata)
             arrays.append(arr.astype(dtype))
         names = [name for name, _, _ in band_spec]
     else:
diff --git a/pyproject.toml b/pyproject.toml
index 772f290..edab353 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dissmodel"
-version = "0.6.1"
+version = "0.6.2"
 description = "Discrete Spatial Modeling framework for raster and vector simulations"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"