Merge pull request #5 from UDST/cache_fix

Enables setting cache=False in size variables
UDST · Feb 17, 2020 · 4d2a57e · 4d2a57e
2 parents cecc3b9 + 99309b5
commit 4d2a57e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/variable_generators/generators.py b/variable_generators/generators.py
@@ -79,13 +79,13 @@ def func():
     return func
 
 
-def make_size_var(agent, geog, geog_id, cache_scope='step', prefix_agent='total'):
+def make_size_var(agent, geog, geog_id, cache=True, cache_scope='step', prefix_agent='total'):
     """
     Generator function for size variables. Registers with orca.
     """
     var_name = prefix_agent + '_' + agent
 
-    @orca.column(geog, var_name, cache=True, cache_scope=cache_scope)
+    @orca.column(geog, var_name, cache=cache, cache_scope=cache_scope)
     def func():
         agents = orca.get_table(agent)
         print('Calculating number of {} for {}'.format(agent, geog))