| 
					
				 | 
			
			
				@@ -7,7 +7,10 @@ class DoorKeyEnv(MiniGridEnv): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				     """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				     def __init__(self, size=8): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-        super().__init__(grid_size=size, max_steps=4 * size) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        super().__init__( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            grid_size=size, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            max_steps=10*size 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        ) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				     def _gen_grid(self, width, height): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         # Create an empty grid 
			 |