@@ -137,7 +137,7 @@ def get_site_scene(
137137 phiEnd = phiEnd ,
138138 clickable = True ,
139139 tooltip = name ,
140- _meta = [site_idx ],
140+ _meta = [site_idx ] if retain_atom_idx else None ,
141141 )
142142 atoms .append (sphere )
143143
@@ -210,7 +210,9 @@ def get_site_scene(
210210 radius = bond_radius / 2 ,
211211 clickable = True ,
212212 tooltip = name_cyl ,
213- _meta = [site_idx , connected_site .index ],
213+ _meta = [site_idx , connected_site .index ]
214+ if retain_atom_idx
215+ else None ,
214216 )
215217 )
216218 trans_vector = trans_vector + 0.25 * max_radius
@@ -222,7 +224,9 @@ def get_site_scene(
222224 radius = bond_radius ,
223225 clickable = True ,
224226 tooltip = name_cyl ,
225- _meta = [site_idx , connected_site .index ],
227+ _meta = [site_idx , connected_site .index ]
228+ if retain_atom_idx
229+ else None ,
226230 )
227231 bonds .append (cylinder )
228232
@@ -233,7 +237,7 @@ def get_site_scene(
233237 radius = bond_radius ,
234238 clickable = True ,
235239 tooltip = name_cyl ,
236- _meta = [site_idx , connected_site .index ],
240+ _meta = [site_idx , connected_site .index ] if retain_atom_idx else None ,
237241 )
238242 bonds .append (cylinder )
239243 all_positions .append (connected_position .tolist ())
@@ -257,7 +261,7 @@ def get_site_scene(
257261 positionPairs = [[position , bond_midpoint .tolist ()]],
258262 color = color ,
259263 radius = bond_radius ,
260- _meta = [site_idx , connected_site .index ],
264+ _meta = [site_idx , connected_site .index ] if retain_atom_idx else None ,
261265 )
262266 bonds .append (cylinder )
263267 all_positions .append (connected_position .tolist ())
0 commit comments