Merge branch 'develop' into parallelization_polymer

MDAnalysis · Jan 6, 2025 · b7328ba · b7328ba
2 parents dc6e903 + 536a390
commit b7328ba
Show file tree

Hide file tree

Showing 157 changed files with 12,468 additions and 7,429 deletions.
diff --git a/package/CHANGELOG b/package/CHANGELOG
@@ -28,6 +28,7 @@ Fixes
 
 Enhancements
  * Enables parallelization for analysis.polymer.PersistenceLength (Issue #4671)
+ * Addition of 'water' token for water selection (Issue #4839)
  * Enables parallelization for analysis.density.DensityAnalysis (Issue #4677, PR #4729)
  * Enables parallelization for analysis.contacts.Contacts (Issue #4660)
  * Enable parallelization for analysis.nucleicacids.NucPairDist (Issue #4670)

diff --git a/package/MDAnalysis/analysis/__init__.py b/package/MDAnalysis/analysis/__init__.py
@@ -22,31 +22,31 @@
 #
 
 __all__ = [
-    'align',
-    'backends',
-    'base',
-    'contacts',
-    'density',
-    'distances',
-    'diffusionmap',
-    'dihedrals',
-    'distances',
-    'dielectric',
-    'gnm',
-    'hbonds',
-    'helix_analysis',
-    'hole2',
-    'hydrogenbonds',
-    'leaflet',
-    'lineardensity',
-    'msd',
-    'nuclinfo',
-    'nucleicacids',
-    'polymer',
-    'pca',
-    'psa',
-    'rdf',
-    'results',
-    'rms',
-    'waterdynamics',
+    "align",
+    "backends",
+    "base",
+    "contacts",
+    "density",
+    "distances",
+    "diffusionmap",
+    "dihedrals",
+    "distances",
+    "dielectric",
+    "gnm",
+    "hbonds",
+    "helix_analysis",
+    "hole2",
+    "hydrogenbonds",
+    "leaflet",
+    "lineardensity",
+    "msd",
+    "nuclinfo",
+    "nucleicacids",
+    "polymer",
+    "pca",
+    "psa",
+    "rdf",
+    "results",
+    "rms",
+    "waterdynamics",
 ]
diff --git a/package/MDAnalysis/analysis/backends.py b/package/MDAnalysis/analysis/backends.py
@@ -35,6 +35,7 @@
 -------
 
 """
+
 import warnings
 from typing import Callable
 from MDAnalysis.lib.util import is_installed
@@ -102,8 +103,9 @@ def _get_checks(self):
             checked during ``_validate()`` run
         """
         return {
-            isinstance(self.n_workers, int) and self.n_workers > 0:
-            f"n_workers should be positive integer, got {self.n_workers=}",
+            isinstance(self.n_workers, int)
+            and self.n_workers
+            > 0: f"n_workers should be positive integer, got {self.n_workers=}",
         }
 
     def _get_warnings(self):
@@ -183,8 +185,8 @@ def _get_warnings(self):
             checked during ``_validate()`` run
         """
         return {
-            self.n_workers == 1:
-            "n_workers is ignored when executing with backend='serial'"
+            self.n_workers
+            == 1: "n_workers is ignored when executing with backend='serial'"
         }
 
     def apply(self, func: Callable, computations: list) -> list:
@@ -307,10 +309,12 @@ def apply(self, func: Callable, computations: list) -> list:
         import dask
 
         computations = [delayed(func)(task) for task in computations]
-        results = dask.compute(computations,
-                               scheduler="processes",
-                               chunksize=1,
-                               num_workers=self.n_workers)[0]
+        results = dask.compute(
+            computations,
+            scheduler="processes",
+            chunksize=1,
+            num_workers=self.n_workers,
+        )[0]
         return results
 
     def _get_checks(self):
@@ -326,8 +330,9 @@ def _get_checks(self):
         """
         base_checks = super()._get_checks()
         checks = {
-            is_installed("dask"):
-            ("module 'dask' is missing. Please install 'dask': "
-             "https://docs.dask.org/en/stable/install.html")
+            is_installed("dask"): (
+                "module 'dask' is missing. Please install 'dask': "
+                "https://docs.dask.org/en/stable/install.html"
+            )
         }
         return base_checks | checks