From 828058728cdba4cc0150ffaf535d49499e0efaf8 Mon Sep 17 00:00:00 2001 From: haixuanTao Date: Thu, 21 Nov 2024 08:46:14 +0100 Subject: [PATCH] Small fix --- examples/piper/record.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/examples/piper/record.py b/examples/piper/record.py index d0d3348e..eff1fb68 100644 --- a/examples/piper/record.py +++ b/examples/piper/record.py @@ -104,7 +104,8 @@ STATE_VEC_LEN = 128 now = datetime.datetime.now() -DATA_DIR = now.strftime("%Y.%m.%d") +ROOT_DIR = os.getenv("DATA_DIR", "/home/agilex/Desktop") +DATA_DIR = ROOT_DIR + now.strftime("%Y.%m.%d") os.makedirs(DATA_DIR, exist_ok=True) ## Make data dir if it does not exist @@ -201,7 +202,7 @@ for event in node: value = event["value"].to_numpy() euler = value[None, 3:6] # Add batch dimension rotmat = convert_euler_to_rotation_matrix(euler) - ortho6d = compute_ortho6d_from_rotation_matrix(rotmat) + ortho6d = compute_ortho6d_from_rotation_matrix(rotmat)[0] values = np.array( [ value[0],