diff options
Diffstat (limited to 'data')
| -rw-r--r-- | data/transformers.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 = [] |