Standardise mapit.MapIt Python imports

mysociety · Sep 24, 2024 · 1de02a9 · 1de02a9
1 parent 81811fc
commit 1de02a9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/hub/management/commands/import_areas.py b/hub/management/commands/import_areas.py
@@ -5,7 +5,7 @@
 from tqdm import tqdm
 
 from hub.models import Area, AreaType
-from utils import mapit
+from utils.mapit import MapIt
 
 
 class Command(BaseCommand):
@@ -58,7 +58,7 @@ def add_arguments(self, parser):
         )
 
     def handle(self, quiet: bool = False, diagnostics: bool = False, *args, **options):
-        mapit_client = mapit.MapIt()
+        mapit_client = MapIt()
         for b_type in self.boundary_types:
             areas = mapit_client.areas_of_type(b_type["mapit_type"], generation=b_type["mapit_generation"])
             if diagnostics:

diff --git a/hub/management/commands/import_council_type.py b/hub/management/commands/import_council_type.py
@@ -3,7 +3,7 @@
 import pandas as pd
 
 from hub.models import DataSet
-from utils import mapit
+from utils.mapit import MapIt
 
 from .base_importers import BaseImportFromDataFrameCommand, MultipleAreaTypesMixin
 
@@ -101,7 +101,7 @@ class Command(MultipleAreaTypesMixin, BaseImportFromDataFrameCommand):
     }
 
     def get_dataframe(self):
-        mapit_client = mapit.MapIt()
+        mapit_client = MapIt()
         areas = mapit_client.areas_of_type(
             self.mapit_types[self.area_type]["mapit_type"],
             generation=self.mapit_types[self.area_type]["mapit_generation"]