diff --git a/demos/franka_pick_place_demo_script.py b/demos/franka_pick_place_demo_script.py index e5abbc1..4cb9ebb 100755 --- a/demos/franka_pick_place_demo_script.py +++ b/demos/franka_pick_place_demo_script.py @@ -123,7 +123,9 @@ def main(): # 2. Extract its directory #script_dir = os.path.dirname(script_path) - script_dir = '/home/student/data/franka_baselines/demos/pick_n_place' # Assumes data folder in user directory. + #script_dir = '/home/student/data/franka_baselines/demos/pick_n_place' # Assumes data folder in user directory. + script_dir = os.path.expanduser('~/data/franka_baselines/demos/pick_n_place') + os.makedirs(script_dir, exist_ok=True) # 3. Create output filename with configuration details fileName = "data_" + robot @@ -404,4 +406,4 @@ def pick_and_place_demo(env, lastObs): deactivate_weld(env, constraint_name="grasp_weld") if __name__ == "__main__": - main() \ No newline at end of file + main()