Simplify load_all

RobertLucey · Jan 1, 2022 · 59552d0 · 59552d0
1 parent 6826c03
commit 59552d0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/road_collisions_uk/bin/load.py b/src/road_collisions_uk/bin/load.py
@@ -5,7 +5,7 @@
 
 def main():
  print('NOTE: Since UK data is so large, only loading data from 2020')
- collisions = Collisions.load_all(region='uk', year=2020)
+ collisions = Collisions.load_all(year=2020)
 
  logger.info('Loaded %s collisions', (len(collisions)))
  logger.info('Do something with the data in the variable \'collisions\'...')

diff --git a/src/road_collisions_uk/models/collision.py b/src/road_collisions_uk/models/collision.py
@@ -161,11 +161,11 @@ def filter(self, **kwargs):
  )
 
  @staticmethod
- def load_all(region=None, year=None):
+ def load_all(year=None):
  import road_collisions_uk
  return Collisions.from_dir(
  '/opt/road_collisions/',
- region=region,
+ region='uk',
  year=year
  )