| 
						
						
							
								
							
						
						
					 | 
				
				 | 
				
					@ -10,7 +10,7 @@ class MiniGridSbShieldingWrapper(gym.core.Wrapper): | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					    def __init__(self, | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					                 env, | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					                 shield_handler : MiniGridShieldHandler, | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					                 create_shield_at_reset = True, | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					                 create_shield_at_reset = False, | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					                 ): | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        super().__init__(env) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self.shield_handler = shield_handler | 
				
			
			
		
	
	
		
			
				
					| 
						
							
								
							
						
						
						
					 | 
				
				 | 
				
					
  |