You are working on the capytaine/capytaine repository. The source code is checked out at /testbed at commit…

You are working on the capytaine/capytaine repository. The source code is checked out at /testbed at commit…: a task in SWE-Lego-RL-2699 (Harbor dataset). data = solver.fill dataset(test matrix, body, hydrostatics=True) RAO = cpt.post pro.rao(data) Right now, user needs to call rao independently…

The task

data = solver.fill_dataset(test_matrix, body, hydrostatics=True) RAO = cpt.post_pro.rao(data) ``` Right now, user needs to call `rao` independently for each water depth: ```python for wd in data.water_depth: RAO = cpt.post_pro.rao(data.sel(water_depth=wd)) ```

Part of Lego-X/Lego-RL-2699.