Update pymol script to use rainbow output

This commit is contained in:
Kevin Wu
2023-02-04 18:08:55 -08:00
parent e27d8b581f
commit 19525239da

View File

@@ -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