credsweeper 1.11.1__py3-none-any.whl → 1.11.3__py3-none-any.whl

credsweeper/ml_model/features/word_in_preamble.py

@@ -1,5 +1,3 @@
-from typing import List
-
 import numpy as np
 
 from credsweeper.common.constants import ML_HUNK
@@ -10,15 +8,6 @@ from credsweeper.ml_model.features.word_in import WordIn
 class WordInPreamble(WordIn):
     """Feature is true if line contains at least one word from predefined list."""
 
-    def __init__(self, words: List[str]) -> None:
-        """Feature returns array of matching words
-
-        Args:
-            words: list of predefined words - MUST BE IN LOWER CASE
-
-        """
-        super().__init__(words)
-
     def extract(self, candidate: Candidate) -> np.ndarray:
         """Returns true if any words in line before variable or value"""
         if 0 <= candidate.line_data_list[0].variable_start:

credsweeper/ml_model/features/word_in_transition.py

@@ -1,5 +1,3 @@
-from typing import List
-
 import numpy as np
 
 from credsweeper.credentials import Candidate
@@ -9,15 +7,6 @@ from credsweeper.ml_model.features.word_in import WordIn
 class WordInTransition(WordIn):
     """Feature is true if line contains at least one word from predefined list."""
 
-    def __init__(self, words: List[str]) -> None:
-        """Feature returns array of matching words
-
-        Args:
-            words: list of predefined words - MUST BE IN LOWER CASE
-
-        """
-        super().__init__(words)
-
     def extract(self, candidate: Candidate) -> np.ndarray:
         """Returns true if any words between variable and value"""
         if 0 <= candidate.line_data_list[0].variable_end < candidate.line_data_list[0].value_start:

credsweeper/ml_model/features/word_in_value.py

@@ -1,5 +1,3 @@
-from typing import List
-
 import numpy as np
 
 from credsweeper.credentials import Candidate
@@ -9,15 +7,6 @@ from credsweeper.ml_model.features.word_in import WordIn
 class WordInValue(WordIn):
     """Feature returns true if candidate value contains at least one word from predefined list."""
 
-    def __init__(self, words: List[str]) -> None:
-        """Feature is true if candidate value contains at least one predefined word.
-
-        Args:
-            words: list of predefined words - MUST BE IN LOWER CASE and SORTED (preferred)
-
-        """
-        super().__init__(words)
-
     def extract(self, candidate: Candidate) -> np.ndarray:
         """Returns array of matching words for first line"""
         if value := candidate.line_data_list[0].value:

credsweeper/ml_model/features/word_in_variable.py

@@ -1,5 +1,3 @@
-from typing import List
-
 import numpy as np
 
 from credsweeper.credentials import Candidate
@@ -9,15 +7,6 @@ from credsweeper.ml_model.features.word_in import WordIn
 class WordInVariable(WordIn):
     """Feature returns array of words matching in variable"""
 
-    def __init__(self, words: List[str]) -> None:
-        """Feature is true if candidate value contains at least one predefined word.
-
-        Args:
-            words: list of predefined words - MUST BE IN LOWER CASE
-
-        """
-        super().__init__(words)
-
     def extract(self, candidate: Candidate) -> np.ndarray:
         """Returns array of matching words for first line"""
         if variable := candidate.line_data_list[0].variable:

credsweeper/ml_model/ml_validator.py

@@ -92,8 +92,9 @@ class MlValidator:
             try:
                 feature = feature_constructor(**kwargs)
             except TypeError:
-                raise TypeError(f'Error while parsing model details. Cannot create feature "{feature_class}"'
-                                f' with kwargs "{kwargs}"')
+                logger.error(f'Error while parsing model details. Cannot create feature "{feature_class}"'
+                             f' with kwargs "{kwargs}"')
+                raise
             if feature_definition["type"] in ["RuleName"]:
                 self.unique_feature_list.append(feature)
             else:
@@ -229,7 +230,7 @@ class MlValidator:
         features_list = []
         probability: np.ndarray = np.zeros(len(group_list), dtype=np.float32)
         head = tail = 0
-        for group_key, candidates in group_list:
+        for _group_key, candidates in group_list:
             line_input, variable_input, value_input, feature_array = self.get_group_features(candidates)
             line_input_list.append(line_input)
             variable_input_list.append(variable_input)