format

hyperon_das/tokenizers/dict_query_tokenizer.py

@@ -1,14 +1,14 @@
 from typing import Any, Callable, Type, TypeAlias
 
 from hyperon_das.tokenizers.elements import (
+    TOKENS_DELIMITER,
+    AndOperator,
     ElementBuilder,
-    Node,
-    Variable,
     Link,
-    OrOperator,
-    AndOperator,
+    Node,
     NotOperator,
-    TOKENS_DELIMITER,
+    OrOperator,
+    Variable,
 )
 
 
@@ -188,6 +188,6 @@ def untokenize(query_tokens: str) -> Query:
             if to_query_callback := DictQueryTokenizer.to_query_mapping.get(type(element)):
                 query = to_query_callback(element)
                 if cursor != len(tokens):
-                    raise ValueError(f"Wrong elements count")
+                    raise ValueError("Wrong elements count")
                 return query
         raise ValueError(f"Unsupported sequence of tokens: {tokens[cursor:]}")

hyperon_das/tokenizers/elements.py

@@ -1,5 +1,5 @@
-from abc import ABC, abstractmethod
 import dataclasses
+from abc import ABC, abstractmethod
 from typing import Any, Type, TypeAlias
 
 TOKENS_DELIMITER = " "

tests/unit/test_dict_query_tokenizer.py

@@ -1,5 +1,7 @@
 import unittest
+
 import pytest
+
 from hyperon_das.tokenizers.dict_query_tokenizer import DictQueryTokenizer
 
 

tests/unit/test_elements.py

@@ -1,13 +1,15 @@
 import unittest
+
 import pytest
+
 from hyperon_das.tokenizers.elements import (
-    Node,
-    Variable,
-    Link,
-    OrOperator,
     AndOperator,
-    NotOperator,
     ElementBuilder,
+    Link,
+    Node,
+    NotOperator,
+    OrOperator,
+    Variable,
 )