Split transactron core

coreblocks/structs_common/rs.py

@@ -4,7 +4,7 @@
 from amaranth.lib.coding import PriorityEncoder
 from transactron import Method, def_method, TModule
 from coreblocks.params import RSLayouts, GenParams, OpType
-from transactron.core import RecordDict
+from transactron.utils import RecordDict
 from transactron.utils.transactron_helpers import make_layout
 
 __all__ = ["RS"]

test/transactions/test_branches.py

@@ -1,15 +1,15 @@
 from amaranth import *
 from itertools import product
 from transactron.core import (
-    CtrlPath,
-    MethodMap,
     TModule,
     Method,
     Transaction,
     TransactionManager,
     TransactionModule,
     def_method,
 )
+from transactron.core.tmodule import CtrlPath
+from transactron.core.manager import MethodMap
 from unittest import TestCase
 from transactron.testing import TestCaseWithSimulator
 from transactron.utils.dependencies import DependencyContext

test/transactions/test_transaction_lib.py

@@ -10,10 +10,9 @@
 
 from amaranth import *
 from transactron import *
-from transactron.core import RecordDict
 from transactron.lib import *
 from coreblocks.utils import *
-from transactron.utils._typing import ModuleLike, MethodStruct
+from transactron.utils._typing import ModuleLike, MethodStruct, RecordDict
 from transactron.utils import ModuleConnector
 from transactron.testing import (
     SimpleTestCircuit,

test/transactions/test_transactions.py

@@ -17,10 +17,10 @@
 
 from transactron.core import (
     Priority,
-    TransactionScheduler,
     trivial_roundrobin_cc_scheduler,
     eager_deterministic_cc_scheduler,
 )
+from transactron.core.manager import TransactionScheduler
 from transactron.utils.dependencies import DependencyContext
 
 

transactron/__init__.py

@@ -1,10 +1 @@
-from .core import *
-
-__all__ = [
-    "TModule",
-    "TransactionManager",
-    "TransactionModule",
-    "Transaction",
-    "Method",
-    "def_method",
-]
+from .core import *  # noqa: F401