add threads_per_worker = 1 in dask clients (#488)

COSIMA · Feb 10, 2025 · 29f9f9c · 29f9f9c
1 parent f3f8c08
commit 29f9f9c
Show file tree

Hide file tree

Showing 25 changed files with 1,850 additions and 328 deletions.
diff --git a/Recipes/Along-slope-velocities.ipynb b/Recipes/Along-slope-velocities.ipynb
@@ -565,7 +565,7 @@
     }
    ],
    "source": [
-    "client = Client() \n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },

diff --git a/Recipes/Atlantic_IndoPacific_Basin_Overturning_Circulation.ipynb b/Recipes/Atlantic_IndoPacific_Basin_Overturning_Circulation.ipynb
@@ -564,7 +564,7 @@
     }
    ],
    "source": [
-    "client = Client()\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -954,7 +954,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:analysis3-24.07]",
+   "display_name": "Python [conda env:analysis3-24.07] *",
    "language": "python",
    "name": "conda-env-analysis3-24.07-py"
   },
@@ -968,7 +968,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.10.16"
   }
  },
  "nbformat": 4,

diff --git a/Recipes/Barotropic_Streamfunction.ipynb b/Recipes/Barotropic_Streamfunction.ipynb
@@ -513,7 +513,7 @@
     }
    ],
    "source": [
-    "client = dask.Client()\n",
+    "client = dask.Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -796,7 +796,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:analysis3-24.01] *",
+   "display_name": "Python [conda env:analysis3-24.01]",
    "language": "python",
    "name": "conda-env-analysis3-24.01-py"
   },

diff --git a/Recipes/Compare_SSH_model_obs.ipynb b/Recipes/Compare_SSH_model_obs.ipynb
@@ -533,7 +533,7 @@
     }
    ],
    "source": [
-    "client = Client()\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -1472,7 +1472,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:analysis3-24.04] *",
+   "display_name": "Python [conda env:analysis3-24.04]",
    "language": "python",
    "name": "conda-env-analysis3-24.04-py"
   },

diff --git a/Recipes/Compare_SST_SSS_TemperatureSalinity_to_WOA13.ipynb b/Recipes/Compare_SST_SSS_TemperatureSalinity_to_WOA13.ipynb
@@ -496,7 +496,7 @@
     }
    ],
    "source": [
-    "client = Client()\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -1708,7 +1708,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.12"
+   "version": "3.10.13"
   }
  },
  "nbformat": 4,

diff --git a/Recipes/Cross-contour_transport.ipynb b/Recipes/Cross-contour_transport.ipynb
@@ -538,7 +538,7 @@
     }
    ],
    "source": [
-    "client = Client()\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },

diff --git a/Recipes/Cross-slope_section.ipynb b/Recipes/Cross-slope_section.ipynb
@@ -550,7 +550,7 @@
     }
    ],
    "source": [
-    "client = Client() \n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },

diff --git a/Recipes/Eddy-Mean_Kinetic_Energy_Decomposition.ipynb b/Recipes/Eddy-Mean_Kinetic_Energy_Decomposition.ipynb
@@ -1496,7 +1496,7 @@
     }
    ],
    "source": [
-    "client = Client(threads_per_worker=1)\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -1937,7 +1937,7 @@
    }
   },
   "kernelspec": {
-   "display_name": "Python [conda env:analysis3-24.04] *",
+   "display_name": "Python [conda env:analysis3-24.04]",
    "language": "python",
    "name": "conda-env-analysis3-24.04-py"
   },

diff --git a/Recipes/Equatorial_thermal_and_zonal_velocity_structure.ipynb b/Recipes/Equatorial_thermal_and_zonal_velocity_structure.ipynb
@@ -63,11 +63,13 @@
     "import numpy as np\n",
     "import xarray as xr\n",
     "import sys, os\n",
+    "from dask.distributed import Client\n",
+    "\n",
     "import warnings\n",
     "warnings.simplefilter(\"ignore\") \n",
+    "\n",
     "import logging\n",
-    "logging.getLogger('flox').setLevel(logging.WARNING)\n",
-    "from dask.distributed import Client"
+    "logging.getLogger('flox').setLevel(logging.WARNING)"
    ]
   },
   {
@@ -2374,7 +2376,7 @@
     }
    ],
    "source": [
-    "client = Client(threads_per_worker=1)\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -3239,7 +3241,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:analysis3-24.07]",
+   "display_name": "Python [conda env:analysis3-24.07] *",
    "language": "python",
    "name": "conda-env-analysis3-24.07-py"
   },
@@ -3253,7 +3255,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.10.16"
   },
   "name": "Equatorial_Pacific.ipynb"
  },

diff --git a/Recipes/Extract_Variables_at_Ocean_Bottom.ipynb b/Recipes/Extract_Variables_at_Ocean_Bottom.ipynb
@@ -2343,7 +2343,7 @@
     }
    ],
    "source": [
-    "client = Client(threads_per_worker=1)\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -6646,7 +6646,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:analysis3-24.04] *",
+   "display_name": "Python [conda env:analysis3-24.04]",
    "language": "python",
    "name": "conda-env-analysis3-24.04-py"
   },

diff --git a/Recipes/Geostrophic_Velocities_from_Sea_Level.ipynb b/Recipes/Geostrophic_Velocities_from_Sea_Level.ipynb
@@ -1452,7 +1452,7 @@
     }
    ],
    "source": [
-    "client = Client(threads_per_worker=1)\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },

diff --git a/Recipes/Hovmoller_Temperature_Depth.ipynb b/Recipes/Hovmoller_Temperature_Depth.ipynb
@@ -2332,7 +2332,7 @@
     }
    ],
    "source": [
-    "client = Client(threads_per_worker=1)\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -2587,7 +2587,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.10.16"
   }
  },
  "nbformat": 4,

diff --git a/Recipes/Meridional_Overturning_Circulation.ipynb b/Recipes/Meridional_Overturning_Circulation.ipynb
@@ -1437,7 +1437,7 @@
     }
    ],
    "source": [
-    "client = Client(threads_per_worker=1)\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },
@@ -2034,7 +2034,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python [conda env:analysis3-24.01] *",
+   "display_name": "Python [conda env:analysis3-24.01]",
    "language": "python",
    "name": "conda-env-analysis3-24.01-py"
   },

diff --git a/Recipes/Meridional_heat_transport.ipynb b/Recipes/Meridional_heat_transport.ipynb
@@ -520,7 +520,7 @@
    ],
    "source": [
     "from dask.distributed import Client\n",
-    "client = Client()\n",
+    "client = Client(threads_per_worker = 1)\n",
     "client"
    ]
   },