Add import select_autoescape (#4327)

Minor refactor in benchcomp

By submitting this pull request, I confirm that my contribution is made
under the terms of the Apache 2.0 and MIT licenses.

Author: zhassan-aws

tools/benchcomp/benchcomp/visualizers/__init__.py

@@ -17,6 +17,8 @@
 import benchcomp
 import benchcomp.visualizers.utils as viz_utils
 
+from jinja2 import Environment, select_autoescape
+
 
 
 @dataclasses.dataclass
@@ -419,8 +421,8 @@ def __call__(self, results):
             "scaled_variants": self._get_scaled_variants(scaled),
         }
 
-        env = jinja2.Environment(
-            loader=jinja2.BaseLoader, autoescape=jinja2.select_autoescape(
+        env = Environment(
+            loader=jinja2.BaseLoader, autoescape=select_autoescape(
                 enabled_extensions=("html"),
                 default_for_string=True))
         env.filters["safe_round"] = self._safe_round