add categories to AccessModel

GeoDaCenter · Aug 6, 2019 · 4709684 · 4709684
1 parent ac52b6c
commit 4709684
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/spatial_access/Models.py b/spatial_access/Models.py
@@ -456,9 +456,17 @@ class AccessModel(ModelData):
     nonprofit services in urban environments.
     """
 
-    def __init__(self, network_type, sources_filename=None, source_column_names=None,
-                 destinations_filename=None, dest_column_names=None, transit_matrix_filename=None,
-                 decay_function='linear', configs=None, debug=False):
+    def __init__(self,
+        network_type,
+        sources_filename=None,
+        source_column_names=None,
+        destinations_filename=None,
+        dest_column_names=None,
+        transit_matrix_filename=None,
+        decay_function='linear', 
+        categories=None,
+        configs=None,
+        debug=False):
         """
         Args:
             network_type: string, one of {'walk', 'bike', 'drive', 'otp'}.