From ec71cec20fef8bb2fa8de586f5868ba051aeb669 Mon Sep 17 00:00:00 2001 From: Surojit Chowdhury Date: Fri, 5 Apr 2024 16:52:59 +0100 Subject: [PATCH] fixing column names while parsing deepracer log dataframe for new reward --- deepracer/logs/log_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepracer/logs/log_utils.py b/deepracer/logs/log_utils.py index 13ed6f9..c2a8e63 100644 --- a/deepracer/logs/log_utils.py +++ b/deepracer/logs/log_utils.py @@ -870,7 +870,7 @@ def df_to_params(df_row, waypoints): 'speed': df_row['speed'], 'steps': df_row['steps'], 'progress': df_row['progress'], - 'heading': df_row['yaw'] * 180 / 3.14, + 'heading': df_row['heading'] * 180 / 3.14, #changing yaw to heading as DeepRacerLog class load with column name heading 'closest_waypoints': closest_waypoints, 'steering_angle': df_row['steering_angle'] * 180 / 3.14, 'waypoints': waypoints,