( self, obs_img: torch.tensor, goal_img: torch.tensor, input_goal_mask: torch.tensor = None, )
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected