| 
						
						
							
								
							
						
						
					 | 
				
				 | 
				
					@ -97,7 +97,7 @@ class MiniGridShieldingWrapper(gym.core.Wrapper): | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					            } | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        ) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self.shield_creator = shield_creator | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self.create_shield_at_reset = create_shield_at_reset | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self.create_shield_at_reset = False # TODO | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self.shield = shield_creator.create_shield(env=self.env) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self.mask_actions = mask_actions | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self.shield_query_creator = shield_query_creator | 
				
			
			
		
	
	
		
			
				
					| 
						
							
								
							
						
						
						
					 | 
				
				 | 
				
					
  |