From d093cb72b4caf455bca58cd027d13551f1bcf25a Mon Sep 17 00:00:00 2001 From: Gabriel Guarisa Date: Wed, 25 Oct 2023 14:43:58 -0300 Subject: [PATCH] Fix linter and gh actions --- .github/workflows/checks.yml | 2 +- retrack/engine/runner.py | 2 +- retrack/nodes/lowercase.py | 2 -- retrack/nodes/match.py | 4 ++-- 4 files changed, 4 insertions(+), 6 deletions(-) diff --git a/.github/workflows/checks.yml b/.github/workflows/checks.yml index ec5708a..7d6a410 100644 --- a/.github/workflows/checks.yml +++ b/.github/workflows/checks.yml @@ -16,7 +16,7 @@ jobs: strategy: fail-fast: false matrix: - python-version: [3.7.16, 3.11] + python-version: [3.8, 3.11] poetry-version: [1.4.2] os: [ubuntu-latest] runs-on: ${{ matrix.os }} diff --git a/retrack/engine/runner.py b/retrack/engine/runner.py index 76da78e..8a99544 100644 --- a/retrack/engine/runner.py +++ b/retrack/engine/runner.py @@ -220,7 +220,7 @@ def execute( self.__run_node(node_id) except Exception as e: raise Exception(f"Error running node {node_id} in {self.name}") from e - + if self.states[constants.OUTPUT_REFERENCE_COLUMN].isna().sum() == 0: break diff --git a/retrack/nodes/lowercase.py b/retrack/nodes/lowercase.py index bc57b59..88a6413 100644 --- a/retrack/nodes/lowercase.py +++ b/retrack/nodes/lowercase.py @@ -1,7 +1,5 @@ import typing -import enum - import pandas as pd import pydantic diff --git a/retrack/nodes/match.py b/retrack/nodes/match.py index df76e2f..3c77c49 100644 --- a/retrack/nodes/match.py +++ b/retrack/nodes/match.py @@ -39,8 +39,8 @@ def kind(self) -> NodeKind: def run(self, input_bool: pd.Series) -> typing.Dict[str, pd.Series]: return { - f"output_then_filter": input_bool, - f"output_else_filter": ~input_bool, + "output_then_filter": input_bool, + "output_else_filter": ~input_bool, } def memory_type(self) -> NodeMemoryType: