Commit efdacc3b authored by root's avatar root
Browse files

temp

parent 1afb381d
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ from dataclasses import dataclass, field, fields
@dataclass
class NavigationConfig(BaseConfig):
    """Configuration class for the Navigation environment."""
    resolution: int = 500
    resolution: int = 300
    eval_set: str = 'base'
    exp_name: str = 'test_base'
    down_sample_ratio: float = 1.0
+5 −0
Original line number Diff line number Diff line
@@ -69,7 +69,12 @@ class NavigationEnv(BaseEnv):
        }
        
        # Initialize AI2-THOR controller
        try:
            self.env = ai2thor.controller.Controller(**self.thor_config)
        except Exception as e:
            print(f"Error initializing AI2-THOR: {e}")
            import traceback
            traceback.print_exc()
        
        # Load dataset
        assert config.eval_set in self.ValidEvalSets