added ActorBase to init

dos-group · Jun 21, 2024 · e0219e6 · e0219e6
1 parent 88c3654
commit e0219e6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/vessim/__init__.py b/vessim/__init__.py
@@ -1,12 +1,14 @@
 """A simulator for carbon-aware applications and systems."""
-from vessim.actor import Actor, ComputingSystem
+
+from vessim.actor import ActorBase, Actor, ComputingSystem
 from vessim.controller import Controller, Monitor
 from vessim.cosim import Microgrid, Environment
 from vessim.policy import MicrogridPolicy, DefaultMicrogridPolicy
 from vessim.signal import Signal, HistoricalSignal, MockSignal
 from vessim.storage import Storage, SimpleBattery
 
 __all__ = [
+    "ActorBase",
     "Actor",
     "ComputingSystem",
     "Controller",
@@ -24,6 +26,7 @@
 
 try:
     from vessim.sil import Broker, SilController, WatttimeSignal, get_latest_event  # noqa: F401
+
     __all__.extend(["Broker", "SilController", "WatttimeSignal", "get_latest_event"])
 except ImportError:
     pass