Spaces:
Sleeping
Sleeping
Removed redundant ligand styling
Browse filesI 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 |
|