Commit 11a98ba7 authored by jameskrw's avatar jameskrw
Browse files

minor

parent e28f63b5
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -385,7 +385,7 @@ class NavigationEnv(BaseEnv):
        )
        
    
        return system_prompt() + '\n' + format_prompt_text
        return system_prompt(format=self.config.prompt_format) + '\n' + format_prompt_text
    
    def compute_reward(self):
        """Compute the total reward for the episode.
+1 −1
Original line number Diff line number Diff line
def system_prompt():
def system_prompt(**kwargs):
    return """You are a home robot and perform navigation tasks according to instructions.

Actions you can take: moveahead, moveback, moveright, moveleft, rotateright, rotateleft, lookup, lookdown.