up

server.py

@@ -492,7 +492,7 @@ class GameManager:
                 sigma_data=0.5,
                 sigma_offset_noise=0.05,
                 noise_previous_obs=True,
-                frame_sampling=None
+                frame_sampling=[{"count": 4, "stride": 1}]  # Required for get_frame_indices
             )
             denoiser = Denoiser(denoiser_cfg)
             
@@ -521,17 +521,36 @@ class GameManager:
             )
             rew_end_model = RewEndModel(rew_end_cfg)
             
-            # Initialize the world model environment with all required components
-            self.engine = WorldModelEnv(
-                denoiser=denoiser,
-                upsampler=upsampler,
-                rew_end_model=rew_end_model,
-                spawn_dir=Path("game/spawn"),
-                num_envs=1,
-                seq_length=16,
-                cfg=world_model_env_cfg,
-                return_denoising_trajectory=False
-            )
+            # Add detailed logging for each component before initialization
+            logger.info("Starting WorldModelEnv initialization")
+            logger.info(f"Denoiser config: {denoiser_cfg}")
+            logger.info(f"Upsampler config: {upsampler_cfg}")
+            logger.info(f"Spawn directory exists: {Path('game/spawn').exists()}")
+            logger.info(f"World model config: {world_model_env_cfg}")
+            
+            try:
+                # Make sure the spawn directory exists
+                spawn_dir = Path("game/spawn")
+                if not spawn_dir.exists():
+                    logger.error(f"Spawn directory {spawn_dir} does not exist")
+
+                # Initialize the world model environment with all required components
+                logger.info("Calling WorldModelEnv constructor")
+                self.engine = WorldModelEnv(
+                    denoiser=denoiser,
+                    upsampler=upsampler,
+                    rew_end_model=rew_end_model,
+                    spawn_dir=spawn_dir,
+                    num_envs=1,
+                    seq_length=16,
+                    cfg=world_model_env_cfg,
+                    return_denoising_trajectory=False
+                )
+                logger.info("WorldModelEnv initialization successful")
+            except Exception as e:
+                logger.error(f"Error initializing WorldModelEnv: {str(e)}")
+                logger.error(f"Error details: {traceback.format_exc()}")
+                raise
             logger.info("Initialized World Model Environment")
         except ImportError as e:
             logger.warning(f"{str(e)}")