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

@@ -13,6 +13,8 @@
 import benchcomp
 import benchcomp.visualizers.utils as viz_utils
 
+from jinja2 import Environment, select_autoescape
+
 
 
 @dataclasses.dataclass
@@ -271,8 +273,8 @@ def __call__(self, results):
             "variants": self._get_variants(metrics),
         }
 
-        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))
         template = env.from_string(self._get_template())