| 
					
					
						
							
						
					
					
				 | 
				@ -24,7 +24,7 @@ class MiniGridSbShieldingWrapper(gym.core.Wrapper): | 
			
		
		
	
		
			
				 | 
				 | 
				        try: | 
				 | 
				 | 
				        try: | 
			
		
		
	
		
			
				 | 
				 | 
				            return self.shield[self.env.get_symbolic_state()] | 
				 | 
				 | 
				            return self.shield[self.env.get_symbolic_state()] | 
			
		
		
	
		
			
				 | 
				 | 
				        except: | 
				 | 
				 | 
				        except: | 
			
		
		
	
		
			
				 | 
				 | 
				            return [1.0] * 3 + [1.0] * 4 | 
				 | 
				 | 
				 | 
			
		
		
	
		
			
				 | 
				 | 
				 | 
				 | 
				 | 
				            return [0.0] * 3 + [1.0] * 4 | 
			
		
		
	
		
			
				 | 
				 | 
				
 | 
				 | 
				 | 
				
 | 
			
		
		
	
		
			
				 | 
				 | 
				    def reset(self, *, seed=None, options=None): | 
				 | 
				 | 
				    def reset(self, *, seed=None, options=None): | 
			
		
		
	
		
			
				 | 
				 | 
				        obs, infos = self.env.reset(seed=seed, options=options) | 
				 | 
				 | 
				        obs, infos = self.env.reset(seed=seed, options=options) | 
			
		
		
	
	
		
			
				| 
					
						
							
						
					
					
					
				 | 
				
  |