Step-example retrieval helper
TrajectoryRetriever(database: TrajectoryDatabase, k: int = 2)
retriever.retrieve_for_plan(goal, k=None) retriever.retrieve_for_step(goal, plan, observation, k=None) retriever.get_retrieved_ids() retriever.clear_retrieved() retriever.record_episode_result(success)
record_episode_result