Skip to content

Commit b62c5d4

Browse files
authored
Updates C++ backend and rat_core extension (#23)
* Updates output result * Adds progress event to rat core * Updates cpp * Updates bayes result
1 parent 44c5c02 commit b62c5d4

File tree

4 files changed

+278
-306
lines changed

4 files changed

+278
-306
lines changed

RAT/events.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,24 @@
11
from typing import Callable, Union, List
22
import RAT.rat_core
3-
from RAT.rat_core import EventTypes, PlotEventData
3+
from RAT.rat_core import EventTypes, PlotEventData, ProgressEventData
44

55

6-
def notify(event_type: EventTypes, data: Union[str, PlotEventData]) -> None:
6+
def notify(event_type: EventTypes, data: Union[str, PlotEventData, ProgressEventData]) -> None:
77
"""Calls registered callbacks with the data when event type has
88
been triggered.
99
1010
Parameters
1111
----------
1212
event_type : EventTypes
1313
The event type that was triggered.
14-
data : str or PlotEventData
14+
data : str or PlotEventData or ProgressEventData
1515
The data sent by the event. The message event data is a string.
1616
"""
1717
callbacks = __event_callbacks[event_type]
1818
for callback in callbacks:
1919
callback(data)
2020

21-
def get_event_callback(event_type: EventTypes) -> List[Callable[[Union[str, PlotEventData]], None]]:
21+
def get_event_callback(event_type: EventTypes) -> List[Callable[[Union[str, PlotEventData, ProgressEventData]], None]]:
2222
"""Returns all callbacks registered for the given event type.
2323
2424
Parameters
@@ -28,20 +28,20 @@ def get_event_callback(event_type: EventTypes) -> List[Callable[[Union[str, Plot
2828
2929
Retuns
3030
------
31-
callback : Callable[[Union[str, PlotEventData]], None]
31+
callback : Callable[[Union[str, PlotEventData, ProgressEventData]], None]
3232
The callback for the event type.
3333
"""
3434
return list(__event_callbacks[event_type])
3535

3636

37-
def register(event_type: EventTypes, callback: Callable[[Union[str, PlotEventData]], None]) -> None:
37+
def register(event_type: EventTypes, callback: Callable[[Union[str, PlotEventData, ProgressEventData]], None]) -> None:
3838
"""Registers a new callback for the event type.
3939
4040
Parameters
4141
----------
4242
event_type : EventTypes
4343
The event type to register.
44-
callback : Callable[[Union[str, PlotEventData]], None]
44+
callback : Callable[[Union[str, PlotEventData, ProgressEventData]], None]
4545
The callback for when the event is triggered.
4646
"""
4747
if not isinstance(event_type, EventTypes):
@@ -60,4 +60,4 @@ def clear() -> None:
6060

6161

6262
__event_impl = RAT.rat_core.EventBridge(notify)
63-
__event_callbacks = {EventTypes.Message: set(), EventTypes.Plot: set()}
63+
__event_callbacks = {EventTypes.Message: set(), EventTypes.Plot: set(), EventTypes.Progress: set()}

cpp/RAT

Submodule RAT updated 200 files

0 commit comments

Comments
 (0)