-
Notifications
You must be signed in to change notification settings - Fork 85
Missing the corelation between forward and backward processes #263
Description
🐛 Describe the bug
when
Traceback (most recent call last):
File "/.conda/envs/hta/lib/python3.10/site-packages/pandas/core/indexes/base.py", line 3812, in get_loc
return self._engine.get_loc(casted_key)
File "pandas/_libs/index.pyx", line 167, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/index.pyx", line 196, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/hashtable_class_helper.pxi", line 7088, in pandas._libs.hashtable.PyObjectHashTable.get_item
File "pandas/_libs/hashtable_class_helper.pxi", line 7096, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: 'fwdbwd_index'
Steps to reproduce
class TraceFWDBWDLinkTestCase(unittest.TestCase):
def setUp(self) -> None:
super().setUp()
test_data_path = Path(file).parent.parent.joinpath(
"tests/data/h100/h100_trace.json"
)
print(f'test_data_path: {test_data_path}')
self.trace = Trace(
trace_files={0: str(test_data_path)},
trace_dir="",
)
self.trace.parse_traces()
self.trace.decode_symbol_ids(use_shorten_name=False)
def test_fwdbwd_index_column(self):
self.assertIn(
"fwdbwd_index",
self.trace.get_trace(0).columns,
"fwdbwd_index column not found in trace DataFrame",
)
def test_fwdbwd_symbol_and_id(self):
self.assertIn(
"fwdbwd",
self.trace.symbol_table.sym_index,
"fwdbwd symbol not found in trace symbol table",
)
Expected behavior
As the testcases in "Steps to reproduce", they should be passed
Environment
Linux and Mac, both.
Python version: 3.10
Install HTA from source.
Additional Info
No response
