fix

Samsung · Sep 10, 2024 · fc55ec3 · fc55ec3
1 parent 7b73add
commit fc55ec3
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/.github/workflows/check.yml b/.github/workflows/check.yml
@@ -32,8 +32,8 @@ jobs:
     - name: Check ml_model.onnx integrity
       if: ${{ always() && steps.code_checkout.conclusion == 'success' }}
       run: |
-        md5sum --binary credsweeper/ml_model/ml_config.json | grep 02655134706ac53f6d60618545b4c86f
-        md5sum --binary credsweeper/ml_model/ml_model.onnx | grep 1fa73275c81e21a84b3a251796af005d
+        md5sum --binary credsweeper/ml_model/ml_config.json | grep 49c4352ae9ec82ad432d49d7e51c27f1
+        md5sum --binary credsweeper/ml_model/ml_model.onnx | grep ff66e97c446d0f2bbd8d37b7dfff7361
 
     # # # line ending
 

diff --git a/credsweeper/ml_model/features/char_set.py b/credsweeper/ml_model/features/char_set.py
@@ -1,4 +1,3 @@
-import contextlib
 import string
 from typing import Dict, Set
 

diff --git a/docs/source/credsweeper.ml_model.features.rst b/docs/source/credsweeper.ml_model.features.rst
@@ -52,10 +52,10 @@ credsweeper.ml\_model.features.is\_secret\_numeric module
    :undoc-members:
    :show-inheritance:
 
-credsweeper.ml\_model.features.match\_in\_attribute module
+credsweeper.ml\_model.features.search\_in\_attribute module
 ----------------------------------------------------------
 
-.. automodule:: credsweeper.ml_model.features.match_in_attribute
+.. automodule:: credsweeper.ml_model.features.search_in_attribute
    :members:
    :undoc-members:
    :show-inheritance:

diff --git a/tests/ml_model/test_features.py b/tests/ml_model/test_features.py
@@ -107,5 +107,5 @@ def test_char_set_n(self):
         self.assertFalse(CharSet("ascii_lowercase").extract(Candidate([self.line_data], [], "rule", Severity.MEDIUM)))
 
     def test_char_set_p(self):
-        self.line_data.value=self.line_data.value.replace(' ', '')
+        self.line_data.value = self.line_data.value.replace(' ', '')
         self.assertTrue(CharSet("ascii_lowercase").extract(Candidate([self.line_data], [], "rule", Severity.MEDIUM)))