Minor errors in trogdar release fixed (#376)

* did requested changes

* warnings removed

* reformatted

Co-authored-by: Keith Fung <[email protected]>

Author: soham4abc

src/electionguard/guardian.py

@@ -44,7 +44,6 @@
 from .schnorr import SchnorrProof
 from .tally import CiphertextTally
 from .types import BALLOT_ID, GUARDIAN_ID
-from .utils import get_optional
 
 
 @dataclass
@@ -511,9 +510,11 @@ def publish_joint_key(self) -> Optional[ELECTION_JOINT_PUBLIC_KEY]:
         )
         return elgamal_combine_public_keys(public_keys)
 
-    def share_other_guardian_key(self, guardian_id: GUARDIAN_ID) -> ElectionPublicKey:
-        """Share other guardians keys shared during key ceremony."""
-        return get_optional(self._guardian_election_public_keys.get(guardian_id))
+    def share_other_guardian_key(
+        self, guardian_id: GUARDIAN_ID
+    ) -> Optional[ElectionPublicKey]:
+        """Share other guardians keys shared during key ceremony"""
+        return self._guardian_election_public_keys.get(guardian_id)
 
     def compute_tally_share(
         self, tally: CiphertextTally, context: CiphertextElectionContext

tests/integration/test_end_to_end_election.py

@@ -60,7 +60,10 @@
 
 from electionguardtest.ballot_factory import BallotFactory
 from electionguardtest.election_factory import ElectionFactory, NUMBER_OF_GUARDIANS
-from electionguardtest.identity_encrypt import identity_auxiliary_encrypt
+from electionguardtest.identity_encrypt import (
+    identity_auxiliary_encrypt,
+    identity_auxiliary_decrypt,
+)
 
 RESULTS_DIR = "test-results"
 DEVICES_DIR = path.join(RESULTS_DIR, "devices")
@@ -249,7 +252,7 @@ def step_1_key_ceremony(self) -> None:
                 if designated_guardian.id is not backup_owner.id:
                     verification = (
                         designated_guardian.verify_election_partial_key_backup(
-                            backup_owner.id, identity_auxiliary_encrypt
+                            backup_owner.id, identity_auxiliary_decrypt
                         )
                     )
                     verifications.append(get_optional(verification))