add ocean heat content to FaIR output

kemccusker · Sep 28, 2023 · 52cc465 · 52cc465
1 parent cb98b9e
commit 52cc465
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/src/scmcoat/core.py b/src/scmcoat/core.py
@@ -323,9 +323,23 @@ def run(self, emissda, simid=None, emissions_driven=True, useMultigas=True):#, r
             coords=[years],
             name="temperature",
         )
-        response_ds = xr.merge(
-            [C_xarray, F_xarray, T_xarray]
-        )
+        if len(ret) == 7:
+            ohc_xarray = xr.DataArray(
+                ohc,
+                dims=[
+                    "year",
+                ],
+                coords=[years],
+                name="ocean_heat_content",
+            )
+            response_ds = xr.merge(
+                [C_xarray, F_xarray, T_xarray, ohc_xarray]
+            )
+        else:
+
+            response_ds = xr.merge(
+                [C_xarray, F_xarray, T_xarray]
+            )
         response_ds["simulation"] = self.simid
 
         # TODO Add attributes