run ruff linting

GiacomoPope · Aug 21, 2024 · b20e480 · b20e480
1 parent 425af71
commit b20e480
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 2 deletions.
diff --git a/benchmarks/benchmark_kyber.py b/benchmarks/benchmark_kyber.py
@@ -45,7 +45,7 @@ def benchmark_kyber(Kyber, name, count):
  enc_times.append(time() - t1)
 
  t2 = time()
- dec = Kyber.decaps(sk, c)
+ _ = Kyber.decaps(sk, c)
  dec_times.append(time() - t2)
 
  avg_keygen = sum(keygen_times) / count

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,2 +1,6 @@
 [tool.black]
 line-length = 79
+
+[tool.ruff.lint]
+# Avoid enforcing Ambiguous variable name (`E741`)
+ignore = ["E741"]
diff --git a/src/kyber_py/kyber/__init__.py b/src/kyber_py/kyber/__init__.py
@@ -1 +1,3 @@
 from .default_parameters import Kyber512, Kyber768, Kyber1024
+
+__all__ = [Kyber512, Kyber768, Kyber1024]
diff --git a/src/kyber_py/ml_kem/__init__.py b/src/kyber_py/ml_kem/__init__.py
@@ -1 +1,3 @@
 from .default_parameters import ML_KEM_512, ML_KEM_768, ML_KEM_1024
+
+__all__ = [ML_KEM_512, ML_KEM_768, ML_KEM_1024]
diff --git a/tests/test_polynomial.py b/tests/test_polynomial.py
@@ -1,5 +1,4 @@
 import unittest
-from random import randint
 from kyber_py.polynomials.polynomials import PolynomialRingKyber