Update end_to_end_test_utilities.py

Author: KRRT7

tests/scripts/end_to_end_test_utilities.py

@@ -8,7 +8,11 @@
 from dataclasses import dataclass, field
 from typing import Optional
 import contextlib
-import tomllib
+
+try:
+    import tomllib
+except ImportError:
+    import tomli as tomllib
 
 
 @dataclass