address PR

VUnit · Oct 8, 2023 · 04ebdef · 04ebdef
1 parent 3e37dd9
commit 04ebdef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/vunit/com/codec_vhdl_package.py b/vunit/com/codec_vhdl_package.py
@@ -148,7 +148,7 @@ def _create_enumeration_of_all_msg_types(self):
 
         msg_type_enumeration_types = []
         for record in self.record_types:
-            if record.elements[0].identifier_List[0] == "msg_type":
+            if record.elements[0].identifier_list[0] == "msg_type":
                 msg_type_enumeration_types.append(record.elements[0].subtype_indication.code)
 
         msg_type_enumeration_literals = []
@@ -276,7 +276,7 @@ def _get_records_with_an_initial_msg_type_element(self):
 
         msg_type_record_types = []
         for record in self.record_types:
-            if record.elements[0].identifier_List[0] == "msg_type":
+            if record.elements[0].identifier_list[0] == "msg_type":
                 msg_type_record_types.append(record)
 
         return msg_type_record_types

diff --git a/vunit/test/bench.py b/vunit/test/bench.py
@@ -13,7 +13,8 @@
 import re
 import bisect
 import collections
-from typing import Any, Iterable, List, Union, OrderedDict
+from collections import OrderedDict
+from typing import Any, Iterable, List, Union
 from ..ostools import file_exists
 from ..cached import cached
 from ..vhdl_parser import remove_comments as remove_vhdl_comments