completed tutorial for the cross-correlation search

mjo22 · May 6, 2024 · 50e2ef6 · 50e2ef6
1 parent bdc037b
commit 50e2ef6
Show file tree

Hide file tree

Showing 4 changed files with 612 additions and 635 deletions.
diff --git a/docs/examples/cross-correlation-search.ipynb b/docs/examples/cross-correlation-search.ipynb
diff --git a/docs/examples/pose-inference.ipynb b/docs/examples/pose-inference.ipynb
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -64,5 +64,6 @@ nav:
             - Read a particle stack: 'examples/read-dataset.ipynb'
         - Intermediate:
             - Simulate a batch of images: 'examples/simulate-micrograph.ipynb'
+            - Run a cross-correlation search: 'examples/cross-correlation-search.ipynb'
     - Simulator API:
         - 'api/simulator/scattering_potential.md'
diff --git a/src/cryojax/rotations/_utils.py b/src/cryojax/rotations/_utils.py
@@ -10,9 +10,7 @@ def convert_quaternion_to_euler_angles(
 
     Adapted from https://github.com/chrisflesher/jax-scipy-spatial/.
     """
-    if len(convention) != 3 or not all(
-        [axis in ["x", "y", "z"] for axis in convention]
-    ):
+    if len(convention) != 3 or not all([axis in ["x", "y", "z"] for axis in convention]):
         raise ValueError(
             f"`convention` should be a string of three characters, each "
             f"of which is 'x', 'y', or 'z'. Instead, got '{convention}'"