[python] Add missing type

lablup · Jan 5, 2024 · 615f14d · 615f14d
1 parent 2fd40b0
commit 615f14d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/binding/python/raftify.pyi b/binding/python/raftify.pyi
@@ -112,6 +112,7 @@ class Peers:
     """ """
 
     def __init__(self, peers: dict) -> None: ...
+    def to_dict(self) -> dict[int, str]: ...
     def items(self) -> list[tuple[int, str]]: ...
     def get(self, node_id: int) -> str: ...
     def add_peer(self, node_id: int, addr: str) -> None: ...

diff --git a/binding/python/src/lib.rs b/binding/python/src/lib.rs
@@ -13,6 +13,7 @@ fn raftify(py: Python, m: &PyModule) -> PyResult<()> {
     m.add_class::<bindings::peers::PyPeers>()?;
     m.add_class::<bindings::raft_client::PyRaftServiceClient>()?;
     m.add_class::<bindings::raft_node::PyRaftNode>()?;
+    m.add_class::<bindings::cluster_join_ticket::PyClusterJoinTicket>()?;
 
     m.add_class::<bindings::raft_rs::eraftpb::conf_change_single::PyConfChangeSingle>()?;
     m.add_class::<bindings::raft_rs::eraftpb::conf_change_transition::PyConfChangeTransition>()?;