From 35b4503ddd148b0c937468891dd0a7e9ff1c79f4 Mon Sep 17 00:00:00 2001 From: Alex Auvolat Date: Wed, 6 May 2015 10:12:17 -0400 Subject: Move weights init to config files ; fix s/time/travel_time --- data/transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'data') diff --git a/data/transformers.py b/data/transformers.py index 1cc4834..e3ff7b1 100644 --- a/data/transformers.py +++ b/data/transformers.py @@ -34,7 +34,7 @@ class Select(Transformer): class TaxiGenerateSplits(Transformer): def __init__(self, data_stream, max_splits=-1): super(TaxiGenerateSplits, self).__init__(data_stream) - self.sources = data_stream.sources + ('destination_latitude', 'destination_longitude', 'time') + self.sources = data_stream.sources + ('destination_latitude', 'destination_longitude', 'travel_time') self.max_splits = max_splits self.data = None self.splits = [] -- cgit v1.2.3