Fix types

kuznia-rdzeni · Dec 18, 2024 · bd14cfb · bd14cfb
1 parent e6735ff
commit bd14cfb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/transactron/testing/profiler.py b/transactron/testing/profiler.py
@@ -20,7 +20,7 @@ async def process(sim: ProcessContext) -> None:
             sim.tick()
             .sample(
                 *(
-                    View(transaction_sample_layout, Cat(transaction.request, transaction.runnable, transaction.grant))
+                    View(transaction_sample_layout, Cat(transaction.ready, transaction.runnable, transaction.run))
                     for transaction in method_map.transactions
                 )
             )

diff --git a/transactron/utils/gen.py b/transactron/utils/gen.py
@@ -193,9 +193,9 @@ def collect_transaction_method_signals(
     get_id = IdGenerator()
 
     for transaction in method_map.transactions:
-        request_loc = get_signal_location(transaction.request, name_map)
+        request_loc = get_signal_location(transaction.ready, name_map)
         runnable_loc = get_signal_location(transaction.runnable, name_map)
-        grant_loc = get_signal_location(transaction.grant, name_map)
+        grant_loc = get_signal_location(transaction.run, name_map)
         transaction_signals_location[get_id(transaction)] = TransactionSignalsLocation(
             request_loc, runnable_loc, grant_loc
         )