Fix tests.

coreblocks/params/keys.py

@@ -2,7 +2,8 @@
 from typing import TYPE_CHECKING
 
 from coreblocks.params.dependencies import SimpleKey, UnifierKey
-from transactron.lib import MethodTryProduct, Collector, Method
+from transactron import Method
+from transactron.lib import MethodTryProduct, Collector
 from coreblocks.peripherals.wishbone import WishboneMaster
 
 if TYPE_CHECKING:

transactron/lib/adapters.py

@@ -4,6 +4,7 @@
 from typing import Optional
 
 __all__ = [
+    "AdapterBase",
     "AdapterTrans",
     "Adapter",
 ]

transactron/lib/methods.py

@@ -3,7 +3,7 @@
 from ..core import RecordDict
 from typing import Optional, Callable, Tuple
 from coreblocks.utils import ValueLike, assign, AssignType
-from .transactions import ManyToOneConnectTrans
+from . import transactions
 from .connections import Forwarder
 
 __all__ = [
@@ -260,7 +260,7 @@ def elaborate(self, platform):
 
         m.submodules.forwarder = forwarder = Forwarder(self.method.data_out.layout)
 
-        m.submodules.connect = ManyToOneConnectTrans(
+        m.submodules.connect = transactions.ManyToOneConnectTrans(
             get_results=[get for get in self.method_list], put_result=forwarder.write
         )
 

transactron/lib/transactions.py

@@ -2,7 +2,7 @@
 from ..core import *
 from ..core import RecordDict
 from typing import Optional, Callable
-from .methods import MethodTransformer
+from . import methods
 
 __all__ = [
     "ConnectTrans",
@@ -84,7 +84,7 @@ def __init__(
     def elaborate(self, platform):
         m = TModule()
 
-        m.submodules.transformer = transformer = MethodTransformer(
+        m.submodules.transformer = transformer = methods.MethodTransformer(
             self.method2,
             i_transform=(self.method1.data_out.layout, self.i_fun),
             o_transform=(self.method1.data_in.layout, self.o_fun),