mypy fixes

datadesk · Jul 25, 2023 · 55a7af2 · 55a7af2
1 parent 32e33f3
commit 55a7af2
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -46,3 +46,4 @@ repos:
     -   id: mypy
         additional_dependencies:
           - types-requests
+          - types-retry
diff --git a/Pipfile b/Pipfile
@@ -12,6 +12,7 @@ pre-commit = "*"
 mypy = "*"
 types-requests = "*"
 rich = "*"
+types-retry = "*"
 
 [packages]
 requests = "*"

diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/nasa_wildfires/__init__.py b/nasa_wildfires/__init__.py
@@ -86,9 +86,9 @@ def _get_features(csv_path: str, verbose: bool = False) -> list[geojson.Feature]
 
     # Convert to GeoJSON
     features = []
-    for r in reader:
-        coords = map(float, [r["longitude"], r["latitude"]])
-        f = geojson.Feature(geometry=geojson.Point(coords), properties=r)
+    for row in reader:
+        coords = map(float, [row["longitude"], row["latitude"]])
+        f = geojson.Feature(geometry=geojson.Point(coords), properties=row)
         features.append(f)
 
     # Return the GeoJSON feature list

diff --git a/test.py b/test.py
@@ -8,7 +8,9 @@
 
 from nasa_wildfires import get_modis, get_viirs_noaa, get_viirs_suomi, options
 
-COMBO_LIST = list(itertools.product(options.REGION_DICT.keys(), options.TIME_RANGE_LIST))
+COMBO_LIST = list(
+    itertools.product(options.REGION_DICT.keys(), options.TIME_RANGE_LIST)
+)
 
 
 class NasaWildfiresUnitTest(unittest.TestCase):