Merge remote-tracking branch 'origin/main' into 44-hydra

Author: sgreenbury

notebooks/00_exploration.ipynb

@@ -249,7 +249,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from auto_cast.metrics.spatiotemporal import MSE\n",
+    "from auto_cast.metrics.spatiotemporal_old import MSE\n",
     "\n",
     "assert trues is not None\n",
     "assert preds.shape == trues.shape\n",
@@ -284,7 +284,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from auto_cast.metrics.spatiotemporal import RMSE\n",
+    "from auto_cast.metrics.spatiotemporal_old import RMSE\n",
     "\n",
     "assert trues is not None\n",
     "rmse_error = RMSE()(preds, trues, 2)"

src/auto_cast/encoders/base.py

@@ -58,8 +58,9 @@ def encode_batch(
         encoded_inputs = self.encode(batch)
 
         # Assign output fields to inputs to be encoded identically in this default impl
-        # Use replace to avoid mutating the original batch
-        output_batch = replace(batch, input_fields=batch.output_fields)
+        # Create a new batch with output fields as input fields to prevent mutation
+        output_batch = replace(batch, input_fields=batch.output_fields.clone())
+
         encoded_outputs = self.encode(output_batch)
 
         # Return encoded batch

src/auto_cast/metrics/__init__.py

@@ -1,4 +1,14 @@
-from .spatiotemporal import MAE, MSE, NMAE, NMSE, NRMSE, RMSE, VMSE, VRMSE, LInfinity
+from .spatiotemporal import (
+    MAE,
+    MSE,
+    NMAE,
+    NMSE,
+    NRMSE,
+    RMSE,
+    VMSE,
+    VRMSE,
+    LInfinity,
+)
 
 __all__ = ["MAE", "MSE", "NMAE", "NMSE", "NRMSE", "RMSE", "VMSE", "VRMSE", "LInfinity"]