diff --git a/envs/webshop/env.py b/envs/webshop/env.py index 92b2922..3f4b07c 100644 --- a/envs/webshop/env.py +++ b/envs/webshop/env.py @@ -348,7 +348,7 @@ async def _run(self, action: str): # Add to trajectory in human agent style try: goal = self.webshop_env.get_instruction_text() - except: + except Exception: goal = "Episode completed" trajectory_entry = { @@ -376,7 +376,7 @@ async def _run(self, action: str): # Add error to trajectory try: goal = self.webshop_env.get_instruction_text() - except: + except Exception: goal = "Error occurred" trajectory_entry = {