Problem with ControlTuner

[Night-howl]

I'm facing the error on the branch 0.2-dev

when i run

from autompc.tuning import ControlTuner
from autompc.sysid import MLP

tuner = ControlTuner(surrogate=MLP(system), surrogate_split=0.5, surrogate_tune_horizon=5)

---

Similarly on the main branch i encounter the error below

AttributeError: 'numpy.ndarray' object has no attribute 'obs'

when i run

tuner2 = PipelineTuner(surrogate_factory=MLPFactory(system), surrogate_split=1.0)
controller2, tune_result2 = tuner.run(pipeline_fixed_model, task, surr_trajs, n_iters=50, 
                                      rng=np.random.default_rng(100), truedyn=benchmark.dynamics)

I'd like to kindly seek your assistance of how to fix this