From 4db69c850118f652ded6607a8d1ce2992634daf9 Mon Sep 17 00:00:00 2001
From: Bogdan Opanchuk <bogdan@opanchuk.net>
Date: Tue, 26 Sep 2023 17:57:12 -0700
Subject: [PATCH] Fix Python typing stubs

---
 nucypher-core-python/nucypher_core/__init__.pyi | 2 +-
 nucypher-core-python/nucypher_core/ferveo.pyi   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/nucypher-core-python/nucypher_core/__init__.pyi b/nucypher-core-python/nucypher_core/__init__.pyi
index c9f6968c..eb41e920 100644
--- a/nucypher-core-python/nucypher_core/__init__.pyi
+++ b/nucypher-core-python/nucypher_core/__init__.pyi
@@ -488,7 +488,7 @@ class ThresholdMessageKit:
 
     ciphertext_header: CiphertextHeader
 
-    def decrypt_with_shared_secret(self, shared_secret: SharedSecret):
+    def decrypt_with_shared_secret(self, shared_secret: SharedSecret) -> bytes:
         ...
 
     @staticmethod
diff --git a/nucypher-core-python/nucypher_core/ferveo.pyi b/nucypher-core-python/nucypher_core/ferveo.pyi
index ff17fd19..77c1746e 100644
--- a/nucypher-core-python/nucypher_core/ferveo.pyi
+++ b/nucypher-core-python/nucypher_core/ferveo.pyi
@@ -224,7 +224,7 @@ def encrypt(message: bytes, aad: bytes, dkg_public_key: DkgPublicKey) -> Ciphert
 
 def combine_decryption_shares_simple(
         decryption_shares: Sequence[DecryptionShareSimple],
-) -> bytes:
+) -> SharedSecret:
     ...