Fix datetime parsing in ShardedDataset to correctly handle ISO format without trailing 'Z'

Author: valhassan

geo_deep_learning/datasets/wds_dataset.py

@@ -313,7 +313,7 @@ def _encode_temporal(self, datetime_str: str) -> torch.Tensor:
             if datetime_str.endswith("Z"):
                 datetime_str = datetime_str[:-1] + "+00:00"
 
-            dt = datetime.fromisoformat(datetime_str.replace("Z", "+00:00"))
+            dt = datetime.fromisoformat(datetime_str)
 
             # Week of year (1-52 or 53)
             week = dt.isocalendar().week