eo-tides 0.6.3.dev5__py3-none-any.whl → 0.6.3.dev7__py3-none-any.whl

eo_tides/model.py

@@ -330,6 +330,9 @@ def ensemble_tides(
     x = tide_df.index.get_level_values(level="x")
     y = tide_df.index.get_level_values(level="y")
 
+    # Identify input datatype
+    input_dtype = tide_df.tide_height.dtype
+
     # Load model ranks points and reproject to same CRS as x and y
     model_ranking_cols = [f"rank_{m}" for m in ensemble_models]
     try:
@@ -373,6 +376,7 @@ def ensemble_tides(
         .set_index(["tide_model", "x", "y"])
         .groupby(["x", "y"])
         .rank()
+        .astype("float32")  # use smaller dtype for rankings to save memory
     )
 
     # If no custom ensemble funcs are provided, use a default ensemble
@@ -404,9 +408,12 @@ def ensemble_tides(
 
         # Use weightings to combine multiple models into single ensemble
         ensemble_df = (
-            # Calculate weighted mean and convert back to dataframe
+            # Calculate weighted mean
             grouped.weighted.sum()
             .div(grouped.weights.sum())
+            # Make sure datatype is the same as the input
+            .astype(input_dtype)
+            # Convert to dataframe
             .to_frame("tide_height")
             # Label ensemble model and ensure indexes are in expected order
             .assign(tide_model=ensemble_n)

{eo_tides-0.6.3.dev5.dist-info → eo_tides-0.6.3.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eo-tides
-Version: 0.6.3.dev5
+Version: 0.6.3.dev7
 Summary: Tide modelling tools for large-scale satellite earth observation analysis
 Project-URL: Homepage, https://GeoscienceAustralia.github.io/eo-tides/
 Project-URL: Repository, https://github.com/GeoscienceAustralia/eo-tides

{eo_tides-0.6.3.dev5.dist-info → eo_tides-0.6.3.dev7.dist-info}/RECORD

@@ -1,10 +1,10 @@
 eo_tides/__init__.py,sha256=pGvVlxMKiYjm_273G-oYcOgVuPra7uEdNZv0oN1i69c,1693
 eo_tides/eo.py,sha256=TuFt9SSiO9Z2o8Kr1g-wFPzofp0HTcgkfhT83zu03kc,23983
-eo_tides/model.py,sha256=0KiHLs5ifN3xcCW7OQU6PQsW7Iq84Wdxqxtz-ml_Bws,37450
+eo_tides/model.py,sha256=jRx6HNgTbK0inNqOkVivr52jPMJPWZ85gJeLjY2Vfk8,37697
 eo_tides/stats.py,sha256=ELQpqIH86442IYgjrGrIK3mi0-pu2ZijFw53arA2FYg,23072
 eo_tides/utils.py,sha256=T19OuPLHzaUKcovCVGANvmOiRu-L8VuDXSTzmNlA6Bo,26647
 eo_tides/validation.py,sha256=KP8WLT5z7KLFjQ9oDla7VJOyLQAK4SVbcz2ySAbsbwI,11882
-eo_tides-0.6.3.dev5.dist-info/METADATA,sha256=y9xFA9qK8lAt5IqgcN6SxW-reyVnjFvtB07o-VnMouY,8229
-eo_tides-0.6.3.dev5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-eo_tides-0.6.3.dev5.dist-info/licenses/LICENSE,sha256=owxWsXViCL2J6Ks3XYhot7t4Y93nstmXAT95Zf030Cc,11350
-eo_tides-0.6.3.dev5.dist-info/RECORD,,
+eo_tides-0.6.3.dev7.dist-info/METADATA,sha256=tpme6HofVs4Y_59_Z_iu7W9TPN4_uVNXR6N_yqqmWI0,8229
+eo_tides-0.6.3.dev7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+eo_tides-0.6.3.dev7.dist-info/licenses/LICENSE,sha256=owxWsXViCL2J6Ks3XYhot7t4Y93nstmXAT95Zf030Cc,11350
+eo_tides-0.6.3.dev7.dist-info/RECORD,,