Make a hack slightly less brittle

all-umass · Mar 19, 2020 · 8a997da · 8a997da
1 parent 078bfed
commit 8a997da
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/backend/web_datasets.py b/backend/web_datasets.py
@@ -286,5 +286,6 @@ def _get_composition_filter_html(m, key, full_key):
   for k, m in m.comps.items():
     html = _get_filter_html(m, key + '-' + k, full_key + '-' + k)
     # HACK: insert composition name just inside the existing <div>
-    html_parts.append('%s%s %s' % (html[:5], comp_name, html[5:]))
+    prefix, suffix = html.split('>', 1)
+    html_parts.append('%s>%s %s' % (prefix, comp_name, suffix))
   return html_parts