Merge branch 'dev' of https://github.com/tkalir/anyway into 2648-addi…

…ng_endpoints_for_location_debugging
data-for-change · May 15, 2024 · 5136d44 · 5136d44
2 parents 4ed78d4 + 3bd68be
commit 5136d44
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/anyway/models.py b/anyway/models.py
@@ -903,7 +903,7 @@ def set_critical(
         resolution = BE_CONST.ResolutionCategories(self.resolution)
         end_time = last_accident_date.to_pydatetime().date()
         start_time = datetime.date(end_time.year + 1 - years_before, 1, 1)
-        location_info = LocationInfo()
+        location_info: LocationInfo = {}
         if resolution == BE_CONST.ResolutionCategories.SUBURBAN_ROAD:
             location_info["road1"] = self.road1
             location_info["road_segment_id"] = self.road_segment_id

diff --git a/anyway/widgets/all_locations_widgets/killed_and_injured_count_per_age_group_widget_utils.py b/anyway/widgets/all_locations_widgets/killed_and_injured_count_per_age_group_widget_utils.py
@@ -34,6 +34,8 @@ def filter_and_group_injured_count_per_age_group(
     ) -> Dict[str, Dict[int, int]]:
         start_time = request_params.start_time
         end_time = request_params.end_time
+        cache_key = None #prevent pylint warning
+
         if request_params.resolution == BE_CONST.ResolutionCategories.STREET:
             involve_yishuv_name = request_params.location_info["yishuv_name"]
             street1_hebrew = request_params.location_info["street1_hebrew"]

diff --git a/tests/test_news_flash.py b/tests/test_news_flash.py
@@ -144,6 +144,7 @@ def test_twitter_parse():
         for k in raw_fields:
             assert getattr(actual, k) == getattr(expected, k)
 
+        actual.set_critical()
         actual.accident = classify_tweets(actual.description)
         assert actual.accident == expected.accident