diff --git a/work_trace/wizards/wt_import_timeline_wizard.py b/work_trace/wizards/wt_import_timeline_wizard.py index 7380789d..feddc63a 100644 --- a/work_trace/wizards/wt_import_timeline_wizard.py +++ b/work_trace/wizards/wt_import_timeline_wizard.py @@ -45,6 +45,8 @@ def _activity_type_to_travel_mode(activity_type): return None if t == 'IN_RAIL_VEHICLE': return 'transit' + if t == 'IN_PASSENGER_VEHICLE' or t == 'FLYING': + return 'flying' if t in VEHICLE_ACTIVITIES: return 'driving' if t in WALKING_ACTIVITIES: @@ -57,7 +59,7 @@ def _activity_type_to_travel_mode(activity_type): def _speed_to_travel_mode(max_speed_mph): """Infer travel mode from maximum speed observed along a path.""" if max_speed_mph > 150: - return 'transit' # flight + return 'flying' if max_speed_mph > 15: return 'driving' if max_speed_mph > 3: