add key to get confirmed trip in read_data (#798)

* add key to get confirmed trip in read_data * modify read_data function
e-mission · Feb 20, 2021 · f4ed2a7 · f4ed2a7
1 parent b53f8df
commit f4ed2a7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/emission/analysis/modelling/tour_model/cluster_pipeline.py b/emission/analysis/modelling/tour_model/cluster_pipeline.py
@@ -40,8 +40,8 @@
 """
 
 #read the data from the database. 
-def read_data(uuid=None):
-    trips = esda.get_entries(esda.CLEANED_TRIP_KEY, uuid,
+def read_data(uuid=None,key=esda.CLEANED_TRIP_KEY):
+    trips = esda.get_entries(key, uuid,
                              time_query=None, geo_query=None)
     logging.info("After reading data, returning %s trips" % len(trips))
     return trips