From 60426c41f6eaeb32a2ad900525f930e85d2eb022 Mon Sep 17 00:00:00 2001 From: Lars Briem Date: Thu, 21 Dec 2023 09:21:19 +0100 Subject: [PATCH] Fix error message --- OTAnalytics/plugin_filter/dataframe_filter.py | 4 ++-- .../plugin_prototypes/track_visualization/track_viz.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/OTAnalytics/plugin_filter/dataframe_filter.py b/OTAnalytics/plugin_filter/dataframe_filter.py index 1a72d2723..8202ef5ae 100644 --- a/OTAnalytics/plugin_filter/dataframe_filter.py +++ b/OTAnalytics/plugin_filter/dataframe_filter.py @@ -89,7 +89,7 @@ def test(self, to_test: DataFrame) -> DataFrame: if not list(to_test.index.names) == [track.TRACK_ID, track.OCCURRENCE]: raise ValueError( f"{track.TRACK_ID} and {track.OCCURRENCE} " - "must be index of DataFrame for filtering to worked." + "must be index of DataFrame for filtering to work." ) return to_test[ to_test.index.get_level_values(INDEX_LEVEL_OCCURRENCE) >= self._start_date @@ -116,7 +116,7 @@ def test(self, to_test: DataFrame) -> DataFrame: if not list(to_test.index.names) == [track.TRACK_ID, track.OCCURRENCE]: raise ValueError( f"{track.TRACK_ID} and {track.OCCURRENCE} " - "must be index of DataFrame for filtering to worked." + "must be index of DataFrame for filtering to work." ) return to_test[ to_test.index.get_level_values(INDEX_LEVEL_OCCURRENCE) <= self._end_date diff --git a/OTAnalytics/plugin_prototypes/track_visualization/track_viz.py b/OTAnalytics/plugin_prototypes/track_visualization/track_viz.py index 0ea06e9a1..ac0e19831 100644 --- a/OTAnalytics/plugin_prototypes/track_visualization/track_viz.py +++ b/OTAnalytics/plugin_prototypes/track_visualization/track_viz.py @@ -281,7 +281,7 @@ def get_data(self) -> DataFrame: if not list(data.index.names) == [track.TRACK_ID, track.OCCURRENCE]: raise ValueError( f"{track.TRACK_ID} and {track.OCCURRENCE} " - "must be index of DataFrame for filtering to worked." + "must be index of DataFrame for filtering to work." ) ids = [track_id.id for track_id in self._filter.get_ids()]