test

learning-at-home · Sep 9, 2024 · cda0234 · cda0234
1 parent c8a8aac
commit cda0234
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hivemind/utils/crypto.py b/hivemind/utils/crypto.py
@@ -51,9 +51,8 @@ def sign(self, data: bytes) -> bytes:
         signature = self._private_key.sign(data)
         return base64.b64encode(signature)
 
-    def get_public_key(self) -> RSAPrivateKey:
-        return RSAPrivateKey(self._private_key.public_key())
-
+    def get_public_key(self) -> RSAPublicKey:  # Fix: return RSAPublicKey, not RSAPrivateKey
+        return RSAPublicKey(self._private_key.public_key())  # Return RSAPublicKey
 
     def to_bytes(self) -> bytes:
         return self._private_key.private_bytes(
@@ -72,6 +71,7 @@ def __setstate__(self, state):
         self._private_key = ed25519.Ed25519PrivateKey.from_private_bytes(self._private_key)
 
 
+
 class RSAPublicKey(PublicKey):
     def __init__(self, public_key: ed25519.Ed25519PublicKey):
         self._public_key = public_key