final release fixes

.github/CODEOWNERS

@@ -0,0 +1 @@
+* 	@michal-g

README.md

@@ -1,4 +1,5 @@
 [![CI](https://github.com/zhonge/cryodrgn/actions/workflows/main.yml/badge.svg)](https://github.com/zhonge/cryodrgn/actions/workflows/main.yml)
+[![beta-release](https://github.com/zhonge/cryodrgn/actions/workflows/beta-release.yml/badge.svg)](https://github.com/zhonge/cryodrgn/actions/workflows/beta-release.yml)
 
 # :snowflake::dragon: cryoDRGN: Deep Reconstructing Generative Networks for cryo-EM and cryo-ET heterogeneous reconstruction
 
@@ -18,7 +19,7 @@ For any feedback, questions, or bugs, please file a Github issue or start a Gith
 * improved interface for `graph` and `pc` traversal methods
 * adding `--datadir` to `cryodrgn abinit_homo` for use with .star files
 * fixing various bugs in Jupyter demonstration notebooks
-* support for TestPyPI beta releases via `pip`
+* support for TestPyPI beta release deployments via `pip`
 
 
 ### New in Version 3.2.x
@@ -192,7 +193,7 @@ The official version 1.0 release. This version introduces several new tools for
     (cryodrgn) $ pip install cryodrgn
 
 You can alternatively install a newer, less stable, development version of `cryodrgn` using our beta release channel:
-    `(cryodrgn) $ pip install -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ 'cryodrgn==3.3.0a1.dev4'
+    `(cryodrgn) $ pip install -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ "cryodrgn<=3.3.0" --pre
 
 More installation instructions are found in the [documentation](https://ez-lab.gitbook.io/cryodrgn/installation).
 

cryodrgn/analysis.py

@@ -562,15 +562,17 @@ def plot_projections(imgs, labels=None, max_imgs=25):
     fig, axes = plt.subplots(
         nrows=nrows, ncols=ncols, figsize=(1.5 * ncols, 1.5 * nrows)
     )
-    axes = axes.ravel()
 
-    for i in range(nrows * ncols):
-        axes[i].imshow(imgs[i], cmap="Greys_r")
-        if labels is not None:
-            axes[i].set_title(labels[i])
+    if not isinstance(axes, np.ndarray):
+        axes = np.array([[axes]])
+    if labels is None:
+        labels = [None for _ in axes.ravel()]
 
-    for i in range(nrows * ncols):
-        axes[i].axis("off")
+    for img, ax, lbl in zip(imgs, axes.ravel(), labels):
+        ax.imshow(img, cmap="Greys_r")
+        if lbl is not None:
+            ax.set_title(lbl)
+        ax.axis("off")
 
     plt.tight_layout()
     return fig, axes