set cach default scope to step for size vars

UDST · Oct 26, 2018 · 77ab93e · 77ab93e
1 parent ae9cb34
commit 77ab93e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/variable_generators/generators.py b/variable_generators/generators.py
@@ -76,13 +76,13 @@ def func():
     return func
 
 
-def make_size_var(agent, geog, geog_id):
+def make_size_var(agent, geog, geog_id, cache_scope='step'):
     """
     Generator function for size variables. Registers with orca.
     """
     var_name = 'total_' + agent
 
-    @orca.column(geog, var_name, cache=True, cache_scope='iteration')
+    @orca.column(geog, var_name, cache=True, cache_scope=cache_scope)
     def func():
         agents = orca.get_table(agent)
         print('Calculating number of {} for {}'.format(agent, geog))