From 19525239dad3007822bb2e619206b39311c8cbac Mon Sep 17 00:00:00 2001 From: Kevin Wu Date: Sat, 4 Feb 2023 18:08:55 -0800 Subject: [PATCH] Update pymol script to use rainbow output --- foldingdiff/pymol_vis.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/foldingdiff/pymol_vis.py b/foldingdiff/pymol_vis.py index 145e7f2..6d15c4a 100644 --- a/foldingdiff/pymol_vis.py +++ b/foldingdiff/pymol_vis.py @@ -26,7 +26,7 @@ def pdb2png(pdb_fname: str, png_fname: str) -> str: # https://gist.github.com/bougui505/11401240 pymol.cmd.load(pdb_fname) pymol.cmd.show("cartoon") - pymol.cmd.color("green") + pymol.cmd.spectrum("count", palette="rainbow") pymol.cmd.set("ray_opaque_background", 0) pymol.cmd.png(png_fname, ray=1, dpi=600) pymol.cmd.delete("*") # So we dont' draw multiple images at once