Skip to content

Commit

Permalink
Fix error message
Browse files Browse the repository at this point in the history
  • Loading branch information
briemla committed Dec 21, 2023
1 parent ccc5113 commit 60426c4
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions OTAnalytics/plugin_filter/dataframe_filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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()]
Expand Down

0 comments on commit 60426c4

Please sign in to comment.