aksell commited on
Commit
5f136f2
·
1 Parent(s): 4501102

Removed redundant ligand styling

Browse files

I don't care about setting the color of the carbons
of the ligand separately.

hexviz/🧬Attention_Visualization.py CHANGED
@@ -86,8 +86,6 @@ def get_3dview(pdb):
86
  for att_weight, first, second, _, _, _ in attention_pairs:
87
  stmol.add_cylinder(xyzview, start=first, end=second, cylradius=att_weight, cylColor='red', dashed=False)
88
 
89
- xyzview.addStyle({"elem": "C", "hetflag": True},
90
- {"stick": {"color": "white", "radius": 0.2}})
91
  xyzview.addStyle({"hetflag": True},
92
  {"stick": {"radius": 0.2}})
93
 
 
86
  for att_weight, first, second, _, _, _ in attention_pairs:
87
  stmol.add_cylinder(xyzview, start=first, end=second, cylradius=att_weight, cylColor='red', dashed=False)
88
 
 
 
89
  xyzview.addStyle({"hetflag": True},
90
  {"stick": {"radius": 0.2}})
91