fix flake8

google · Sep 21, 2024 · d412a71 · d412a71
1 parent da401d4
commit d412a71
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 24 deletions.
diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml
@@ -24,11 +24,11 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install -r requirements.txt
-        pip install pylint
         pip install codecov
-    - name: Lint with pylint
+    - name: Lint with flake8
       run: |
-        bash run_pylint.sh
+        pip install flake8
+        flake8 --indent-size 2 --max-line-length 80 **/*.py *.py
     - name: Run tests
       run: |
         bash run_tests.sh

diff --git a/requirements.txt b/requirements.txt
@@ -3,3 +3,4 @@ scipy>=1.1.0
 torch>=1.4.0
 codecov==2.1.13
 coverage==7.6.0
+colortimelog
diff --git a/run_pylint.sh b/run_pylint.sh
diff --git a/tests/uisrnn_test.py b/tests/uisrnn_test.py
@@ -91,7 +91,7 @@ def test_fit_list_and_predict_single_label(self):
     train_cluster_ids = [
         np.array(['A'] * 100),
         np.array(['A'] * 200),
-        np.array(['A'] * 300),]
+        np.array(['A'] * 300)]
 
     model = uisrnn.UISRNN(model_args)
 

diff --git a/uisrnn/__init__.py b/uisrnn/__init__.py
@@ -20,11 +20,9 @@
 
 from . import arguments
 from . import evals
-from . import loss_func
 from . import uisrnn
 from . import utils
 
-#pylint: disable=C0103
 parse_arguments = arguments.parse_arguments
 compute_sequence_match_accuracy = evals.compute_sequence_match_accuracy
 output_result = utils.output_result