Add a jupyter notebook for regenerating and customizing figures #219

ml-struct-bio · Mar 26, 2023 · f44a988 · f44a988
1 parent 521a459
commit f44a988
Show file tree

Hide file tree

Showing 2 changed files with 527 additions and 0 deletions.
diff --git a/cryodrgn/commands/analyze.py b/cryodrgn/commands/analyze.py
@@ -391,6 +391,16 @@ def main(args):
         logger.info(f"{out_ipynb} already exists. Skipping")
     logger.info(out_ipynb)
 
+    # copy over template if file doesn't exist
+    out_ipynb = f"{outdir}/cryoDRGN_figures.ipynb"
+    if not os.path.exists(out_ipynb):
+        logger.info("Creating jupyter notebook...")
+        ipynb = f"{cryodrgn._ROOT}/templates/cryoDRGN_figures_template.ipynb"
+        shutil.copyfile(ipynb, out_ipynb)
+    else:
+        logger.info(f"{out_ipynb} already exists. Skipping")
+    logger.info(out_ipynb)
+
     logger.info(f"Finished in {dt.now()-t1}")