black format

GiacomoPope · Jul 22, 2024 · e69ebb5 · e69ebb5
1 parent 514f884
commit e69ebb5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/kyber/kyber.py b/kyber/kyber.py
@@ -2,6 +2,7 @@
 from hashlib import sha3_256, sha3_512, shake_128, shake_256
 from modules.modules import ModuleKyber
 
+
 class Kyber:
     def __init__(self, parameter_set, seed=None):
         self.k = parameter_set["k"]
@@ -29,11 +30,14 @@ def set_drbg_seed(self, seed):
         """
         try:
             from drbg.aes256_ctr_drbg import AES256_CTR_DRBG
+
             self.drbg = AES256_CTR_DRBG(seed)
             self.random_bytes = self.drbg.random_bytes
         except ImportError as e:
             print(f"Error importing AES from pycryptodome: {e = }")
-            print("Have you tried installing requirements: pip -r install requirements")
+            print(
+                "Have you tried installing requirements: pip -r install requirements"
+            )
 
     def reseed_drbg(self, seed):
         """

diff --git a/ml_kem/ml_kem.py b/ml_kem/ml_kem.py
@@ -2,6 +2,7 @@
 from hashlib import sha3_256, sha3_512, shake_128, shake_256
 from modules.modules import ModuleKyber
 
+
 class ML_KEM:
     def __init__(self, params, seed=None):
         # ml-kem params
@@ -30,11 +31,14 @@ def set_drbg_seed(self, seed):
         """
         try:
             from drbg.aes256_ctr_drbg import AES256_CTR_DRBG
+
             self.drbg = AES256_CTR_DRBG(seed)
             self.random_bytes = self.drbg.random_bytes
         except ImportError as e:
             print(f"Error importing AES from pycryptodome: {e = }")
-            print("Have you tried installing requirements: pip -r install requirements")
+            print(
+                "Have you tried installing requirements: pip -r install requirements"
+            )
 
     def reseed_drbg(self, seed):
         """