Ci improvement

.github/workflows/main.yml

@@ -34,7 +34,19 @@ jobs:
           key: venv-${{ hashFiles('poetry.lock') }}
 
       - name: Install the project dependencies
-        run: poetry install
+        run: poetry install --all-extras
+
+      - name: Check formatting with black
+        run: poetry run black .
+
+      - name: Run type checker
+        run: poetry run mypy --follow-imports=skip --check-untyped-defs .
+
+      - name: Run ruff linter
+        run: poetry run ruff check .
+
+      - name: Run pylint linter
+        run: poetry run pylint knowledge_verificator/
 
       - name: Run the automated test
         run: poetry run pytest -v

knowledge_verificator/nli.py

@@ -1,11 +1,10 @@
 """Natural Language Inference module with pre-trained RoBERTa-Large."""
 
 from enum import Enum
-import operator
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
-import torch
 import warnings
 import logging
+from transformers import AutoTokenizer, AutoModelForSequenceClassification # type: ignore
+import torch
 
 
 class Relation(Enum):
@@ -78,12 +77,11 @@ def infer(
 
     entailment = round(predicted_probability[0], precision)
     neutral = round(predicted_probability[1], precision)
-    contradiction = round(1. - entailment - neutral, precision)
 
     return {
         Relation.ENTAILMENT : entailment,
         Relation.NEUTRAL : neutral,
-        Relation.CONTRADICTION : contradiction,
+        Relation.CONTRADICTION : round(1. - entailment - neutral, precision)
     }
 
 def infer_relation(