modified examples

Hydrology-IFH · Sep 11, 2024 · 99b4a20 · 99b4a20
1 parent 1d9d92e
commit 99b4a20
Show file tree

Hide file tree

Showing 1,087 changed files with 3,479 additions and 3,362 deletions.
diff --git a/examples/plot_scale/boadkh/plot_nitrate_leaching.py b/examples/plot_scale/boadkh/plot_nitrate_leaching.py
diff --git a/examples/plot_scale/boadkh/plot_single_time_series.py b/examples/plot_scale/boadkh/plot_single_time_series.py
@@ -252,7 +252,11 @@ def repeat_by_areashare(values, area_share):
 locations = ["freiburg"]
 vars_sim = ["C_q_ss"]
 crop_rotation_scenarios = ["winter-wheat_silage-corn"]
-fertilization_intensities = ["medium"]
+fertilization_intensities = ["high"]
+
+# load model parameters
+csv_file = base_path / "parameters.csv"
+df_params = pd.read_csv(csv_file, sep=";", skiprows=1)
 
 # load simulated fluxes and states
 dict_fluxes_states = {}
@@ -308,12 +312,11 @@ def repeat_by_areashare(values, area_share):
         for fertilization_intensity in fertilization_intensities:
             for var_sim in vars_sim:
                 ds = dict_fluxes_states[location][crop_rotation_scenario]
-                sim_vals1 = ds["q_ss"].isel(y=0, x=36).values
+                sim_vals1 = ds["q_ss"].isel(y=0, x=36).values[1:]
                 ds = dict_nitrate[location][crop_rotation_scenario][f'{fertilization_intensity}_Nfert'] 
-                sim_vals2 = ds["M_q_ss"].isel(y=0, x=36).values[1:] * 4.427  # convert nitrate-nitrogen to nitrate
-                sim_vals = onp.where(sim_vals1 > 0.1, sim_vals2/sim_vals1, onp.nan)
-                sim_vals = onp.where(sim_vals > 1000, onp.nan, sim_vals)                  
-                df = pd.DataFrame(index=ds["Time"].values[1:], data=sim_vals)
+                sim_vals2 = ds["M_q_ss"].isel(y=0, x=36).values[1:-1] * 4.427  # convert nitrate-nitrogen to nitrate
+                sim_vals = onp.where(sim_vals1 > 0.01, sim_vals2/sim_vals1, onp.nan)
+                df = pd.DataFrame(index=ds["Time"].values[1:-1], data=sim_vals)
                 df = df.loc["2014-01-01":"2023-12-31"]
                 fig, ax = plt.subplots(1, 1, figsize=(6, 2))
                 ax.plot(df.index, df.values, color="black", linewidth=1)
@@ -327,18 +330,68 @@ def repeat_by_areashare(values, area_share):
                 fig.savefig(file, dpi=300)
                 plt.close(fig)
 
+
+for crop_rotation_scenario in crop_rotation_scenarios:
+    for location in locations:
+        for fertilization_intensity in fertilization_intensities:
+            for var_sim in vars_sim:
+                ds = dict_nitrate[location][crop_rotation_scenario][f'{fertilization_intensity}_Nfert'] 
+                sim_vals = ds["C_q_ss"].isel(y=0, x=36).values[1:] * 4.427  # convert nitrate-nitrogen to nitrate
+                df = pd.DataFrame(index=ds["Time"].values[1:], data=sim_vals)
+                df = df.loc["2014-01-01":"2023-12-31"]
+                fig, ax = plt.subplots(1, 1, figsize=(6, 2))
+                ax.plot(df.index, df.values, color="black", linewidth=1)
+                ax.set_xlabel("Time [Year]")
+                ax.set_ylabel(_lab_unit_daily[var_sim])
+                ax.set_xlim(df.index[0], df.index[-1])
+                ax.set_ylim(0, )
+                plt.xticks(rotation=33)
+                fig.tight_layout()
+                file = base_path_figs / f"trace_{var_sim}_{location}_{crop_rotation_scenario}__.png"
+                fig.savefig(file, dpi=300)
+                plt.close(fig)
+
 for crop_rotation_scenario in crop_rotation_scenarios:
     for location in locations:
         for fertilization_intensity in fertilization_intensities:
             for var_sim in vars_sim:
                 fig, ax = plt.subplots(2, 1, figsize=(6, 4), sharex=True)
+                ds = dict_nitrate[location][crop_rotation_scenario][f'{fertilization_intensity}_Nfert'] 
+                sim_vals = ds["C_q_ss"].isel(y=0, x=36).values[1:] * 4.427  # convert nitrate-nitrogen to nitrate
+                # sim_vals = onp.where(sim_vals > 200, onp.nan, sim_vals)                  
+                df = pd.DataFrame(index=ds["Time"].values[1:], data=sim_vals)
+                df = df.loc["2016-01-01":"2017-12-31"]
+                ax[1].plot(df.index, df.values, color="#c994c7", linewidth=1)
+                ax[1].set_xlabel("Time [Year]")
+                ax[1].set_ylabel(_lab_unit_daily[var_sim])
+                ax[1].set_xlim(df.index[0], df.index[-1])
+                ax[1].set_ylim(0, )
                 ds = dict_fluxes_states[location][crop_rotation_scenario]
-                sim_vals1 = ds["q_ss"].isel(y=0, x=36).values
+                sim_vals = ds["q_ss"].isel(y=0, x=36).values[1:]               
+                df = pd.DataFrame(index=ds["Time"].values[1:], data=sim_vals)
+                df = df.loc["2016-01-01":"2017-12-31"]
+                ax[0].plot(df.index, df.values, color="#9ecae1", linewidth=1)
+                ax[0].set_ylabel(_lab_unit_daily["q_ss"])
+                ax[0].set_xlim(df.index[0], df.index[-1])
+                ax[0].set_ylim(0, )
+                plt.xticks(rotation=33)
+                fig.tight_layout()
+                file = base_path_figs / f"trace_perc_nitrate_{location}_{crop_rotation_scenario}_2016_2017_.png"
+                fig.savefig(file, dpi=300)
+                plt.close(fig)
+
+for crop_rotation_scenario in crop_rotation_scenarios:
+    for location in locations:
+        for fertilization_intensity in fertilization_intensities:
+            for var_sim in vars_sim:
+                fig, ax = plt.subplots(2, 1, figsize=(6, 4), sharex=True)
+                ds = dict_fluxes_states[location][crop_rotation_scenario]
+                sim_vals1 = ds["q_ss"].isel(y=0, x=36).values[1:]
                 ds = dict_nitrate[location][crop_rotation_scenario][f'{fertilization_intensity}_Nfert'] 
-                sim_vals2 = ds["M_q_ss"].isel(y=0, x=36).values[1:] * 4.427  # convert nitrate-nitrogen to nitrate
+                sim_vals2 = ds["M_q_ss"].isel(y=0, x=36).values[1:-1] * 4.427  # convert nitrate-nitrogen to nitrate
                 sim_vals = onp.where(sim_vals1 > 0.1, sim_vals2/sim_vals1, onp.nan)
                 # sim_vals = onp.where(sim_vals > 200, onp.nan, sim_vals)                  
-                df = pd.DataFrame(index=ds["Time"].values[1:], data=sim_vals)
+                df = pd.DataFrame(index=ds["Time"].values[1:-1], data=sim_vals)
                 df = df.loc["2016-01-01":"2017-12-31"]
                 ax[1].plot(df.index, df.values, color="#c994c7", linewidth=1)
                 ax[1].set_xlabel("Time [Year]")
@@ -357,4 +410,55 @@ def repeat_by_areashare(values, area_share):
                 fig.tight_layout()
                 file = base_path_figs / f"trace_perc_nitrate_{location}_{crop_rotation_scenario}_2016_2017.png"
                 fig.savefig(file, dpi=300)
-                plt.close(fig)
+                plt.close(fig)
+
+
+colors = sns.color_palette("RdPu", n_colors=len(fertilization_intensities))
+for crop_rotation_scenario in crop_rotation_scenarios:
+    for location in locations:
+        for i, fertilization_intensity in enumerate(fertilization_intensities):
+            ds = dict_fluxes_states[location][crop_rotation_scenario]
+            sim_vals1 = ds["q_ss"].isel(y=0).values[:, 1:]
+            ds = dict_nitrate[location][crop_rotation_scenario][f'{fertilization_intensity}_Nfert'] 
+            sim_vals2 = ds["M_q_ss"].isel(y=0).values[:, 1:-1] * 4.427  # convert nitrate-nitrogen to nitrate
+            sim_vals = onp.where(sim_vals1 > 0.01, sim_vals2/sim_vals1, onp.nan)
+            cond1 = (df_params["CLUST_flag"] == 1)
+            df = pd.DataFrame(index=ds["Time"].values[1:-1], data=sim_vals.T).loc[:, cond1]
+            df = df.loc["2014-01-01":"2022-12-31", :]
+            vals = df.values.T
+            idx = np.argmax(np.nanmax(vals, axis=1))
+
+            fig, ax = plt.subplots(2, 1, figsize=(6, 4), sharex=True)
+            color = "blue"
+            ds = dict_fluxes_states[location][crop_rotation_scenario]
+            sim_vals = ds["q_ss"].isel(y=0).values
+            cond1 = (df_params["CLUST_flag"] == 1)
+            df = pd.DataFrame(index=ds["Time"].values, data=sim_vals.T).loc[:, cond1]
+            df = df.loc["2014-01-01":"2022-12-31", :]
+            vals = df.values.T[idx, :]
+            ax[0].plot(df.index, vals, color=color, label="Median", linewidth=1)
+            ax[0].legend(frameon=False, loc="upper right", ncol=4, bbox_to_anchor=(0.7, 1.3))
+            ax[0].set_ylabel('PERC [mm/day]')
+            ax[0].set_xlim(df.index[0], df.index[-1])
+            ax[0].set_ylim(0, 1)
+
+            color = colors[i]
+            ds = dict_fluxes_states[location][crop_rotation_scenario]
+            sim_vals1 = ds["q_ss"].isel(y=0).values[:, 1:]
+            ds = dict_nitrate[location][crop_rotation_scenario][f'{fertilization_intensity}_Nfert'] 
+            sim_vals2 = ds["M_q_ss"].isel(y=0).values[:, 1:-1] * 4.427  # convert nitrate-nitrogen to nitrate
+            sim_vals = onp.where(sim_vals1 > 0.01, sim_vals2/sim_vals1, onp.nan)
+            cond1 = (df_params["CLUST_flag"] == 1)
+            df = pd.DataFrame(index=ds["Time"].values[1:-1], data=sim_vals.T).loc[:, cond1]
+            df = df.loc["2014-01-01":"2022-12-31", :]
+            vals = df.values.T[idx, :]
+            ax[1].plot(df.index, vals, color=color, label="Median", linewidth=1)
+            ax[1].legend(frameon=False, loc="upper right", ncol=4, bbox_to_anchor=(0.7, 1.3))
+            ax[1].set_xlabel("Time [Year]")
+            ax[1].set_ylabel(_lab_unit_daily["C_q_ss"])
+            ax[1].set_xlim(df.index[0], df.index[-1])
+            ax[1].set_ylim(0, 1000)
+            fig.tight_layout()
+            file = base_path_figs / f"trace_perc_flux_nitrate_conc_{location}_{crop_rotation_scenario}_{fertilization_intensity}_{idx}.png"
+            fig.savefig(file, dpi=300)
+            plt.close(fig)
diff --git a/...t_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_high_Nfert_slurm.sh b/...t_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_high_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_bare-grass.nc | cut -f
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...ot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_low_Nfert_slurm.sh b/...ot_scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_low_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_bare-grass.nc | cut -f
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_medium_Nfert_slurm.sh b/...scale/boadkh/svat_crop_nitrate/svat_crop_nitrate_bretten_bare-grass_medium_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_bare-grass.nc | cut -f
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario bare-grass --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_high_Nfert_slurm.sh b/...crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_high_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/..._crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_low_Nfert_slurm.sh b/..._crop_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_low_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...op_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_medium_Nfert_slurm.sh b/...op_nitrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_clover_medium_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_clover --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...trate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_high_Nfert_slurm.sh b/...trate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_high_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...itrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_low_Nfert_slurm.sh b/...itrate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_low_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...ate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_medium_Nfert_slurm.sh b/...ate/svat_crop_nitrate_bretten_grain-corn_winter-wheat_winter-barley_medium_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity medium -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley --fertilization-intensity medium -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/..._nitrate_bretten_grain-corn_winter-wheat_winter-barley_yellow-mustard_high_Nfert_slurm.sh b/..._nitrate_bretten_grain-corn_winter-wheat_winter-barley_yellow-mustard_high_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity high -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity high -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate

diff --git a/...p_nitrate_bretten_grain-corn_winter-wheat_winter-barley_yellow-mustard_low_Nfert_slurm.sh b/...p_nitrate_bretten_grain-corn_winter-wheat_winter-barley_yellow-mustard_low_Nfert_slurm.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-#SBATCH --time=32:00:00
+#SBATCH --time=72:00:00
 #SBATCH --nodes=1
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=1
@@ -28,7 +28,7 @@ checksum_ssd=$(shasum -a 256 "${TMPDIR}"/SVATCROP_bretten_grain-corn_winter-whea
 done
 echo "Copying was successful"
 
-python svat_crop_nitrate.py -b jax -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity low -td "${TMPDIR}"
+python svat_crop_nitrate.py -d cpu --float-type float64 --location bretten --crop-rotation-scenario grain-corn_winter-wheat_winter-barley_yellow-mustard --fertilization-intensity low -td "${TMPDIR}"
 # Move output from local SSD to global workspace
 echo "Move output to /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate"
 mkdir -p /pfs/work7/workspace/scratch/fr_rs1092-workspace1/roger/examples/plot_scale/boadkh/output/svat_crop_nitrate