Fixing more issues

cog-imperial · Dec 10, 2024 · c58d370 · c58d370
1 parent 7269355
commit c58d370
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/omlt/base/constraint.py b/src/omlt/base/constraint.py
@@ -56,7 +56,7 @@ def __init__(
         self.name = None
         self.format = lang
 
-    def keys(self, *, sort=False):  # noqa: ARG002
+    def keys(self, sort=False):  # noqa: ARG002, FBT002
         yield from self._index_set
 
 

diff --git a/src/omlt/base/julia.py b/src/omlt/base/julia.py
@@ -448,7 +448,7 @@ def __init__(self, *indexes: Any, **kwargs: Any):
         self.format = "jump"
         self._jumpcons = {idx: None for idx in self._index_set[0]}
 
-    def keys(self, *, sort=False):  # noqa: ARG002
+    def keys(self, sort=False):  # noqa: ARG002, FBT002
         yield from self._index_set
 
     def __setitem__(self, label, item):