make max entries per row configurable

Sharma05 · Jul 18, 2020 · e346e31 · e346e31
1 parent 6829749
commit e346e31
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/features/model.py b/features/model.py
@@ -188,7 +188,7 @@ def add_eps(a):
     return a + eps
 
 
-MAX_ENTRIES_PER_ROW = 1664
+MAX_ENTRIES_PER_ROW = int(os.environ.get("MAX_ENTRIES_PER_ROW", "1664"))
 
 def select_feature_matrix(conn, table_name, year, cohort_features, feature_a, feature_b):
     table = tables[table_name]