fix(py): missing ValidatorMessage stub in Python bindings

nucypher · Jun 22, 2023 · 2c3aa01 · 2c3aa01
1 parent 6b33158
commit 2c3aa01
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/nucypher-core-python/nucypher_core/ferveo.py b/nucypher-core-python/nucypher_core/ferveo.py
@@ -10,6 +10,7 @@
 
 Validator = _ferveo.Validator
 Transcript = _ferveo.Transcript
+ValidatorMessage = _ferveo.ValidatorMessage
 Dkg = _ferveo.Dkg
 Ciphertext = _ferveo.Ciphertext
 DecryptionShareSimple = _ferveo.DecryptionShareSimple

diff --git a/nucypher-core-python/nucypher_core/ferveo.pyi b/nucypher-core-python/nucypher_core/ferveo.pyi
@@ -1,4 +1,4 @@
-from typing import Sequence, Tuple
+from typing import Sequence
 
 
 class Keypair:
@@ -65,6 +65,19 @@ class DkgPublicKey:
         ...
 
 
+class ValidatorMessage:
+
+    def __init__(
+            self,
+            validator: Validator,
+            transcript: Transcript,
+    ):
+        ...
+
+    validator: Validator
+    transcript: Transcript
+
+
 class Dkg:
 
     def __init__(
@@ -82,7 +95,7 @@ class Dkg:
     def generate_transcript(self) -> Transcript:
         ...
 
-    def aggregate_transcripts(self, messages: Sequence[Tuple[Validator, Transcript]]) -> AggregatedTranscript:
+    def aggregate_transcripts(self, messages: Sequence[ValidatorMessage]) -> AggregatedTranscript:
         ...
 
 
@@ -115,10 +128,10 @@ class DecryptionSharePrecomputed:
 
 class AggregatedTranscript:
 
-    def __init__(self, messages: Sequence[Tuple[Validator, Transcript]]):
+    def __init__(self, messages: Sequence[ValidatorMessage]):
         ...
 
-    def verify(self, shares_num: int, messages: Sequence[Tuple[Validator, Transcript]]) -> bool:
+    def verify(self, shares_num: int, messages: Sequence[ValidatorMessage]) -> bool:
         ...
 
     def create_decryption_share_simple(
@@ -246,4 +259,4 @@ class ValidatorPublicKeyMismatch(Exception):
 
 
 class SerializationError(Exception):
-    pass
+    pass