Commit f0ef6370 authored by jameskrw's avatar jameskrw
Browse files

Merge branch 'dev' of https://github.com/RAGEN-AI/vagen into dev

parents 367cf380 f993c858
Loading
Loading
Loading
Loading
+16 −0
Original line number Diff line number Diff line
seed: 42
train_path: "./data/sokoban-vision-6-step/train.parquet"
test_path: "./data/sokoban-vision-6-step/test.parquet"
"env1": {
  "env_name": "sokoban",
  'env_config': {
      'num_boxes': 1,
      'dim_room' : [6,6],
      'max_steps': 100,
      'render_mode': "vision",
      'min_actions_to_succeed': 5,
      'max_actions_per_step': 3
  },
  "train_size": 10000,
  "test_size": 1000,
}
 No newline at end of file
+1 −0
Original line number Diff line number Diff line
@@ -101,6 +101,7 @@ class SokobanEnv(BaseEnv):
        info["metrics"] = metrics
        metrics['turn_metrics']['action_is_effective'] = not np.array_equal(prev_player_position, self.env.player_position)
        self.total_reward += self.reward

        return self._render(init_obs=False), self.reward, done, info
    
    def system_prompt(self):
+1 −0
Original line number Diff line number Diff line
# env_utils.py
import os
import random
import logging
import numpy as np