Cleanup

ml-agents/mlagents/trainers/torch/networks.py

 
     def forward(
         self,
-        all_net_inputs: List[List[torch.Tensor]],
+        net_inputs: List[List[torch.Tensor]],
-            all_net_inputs, actions, memories, sequence_length
+            net_inputs, actions, memories, sequence_length
         )
         output = self.value_heads(encoding)
         return output, memories
     @abc.abstractmethod
     def forward(
         self,
-        net_inputs: List[torch.Tensor],
+        vec_inputs: List[torch.Tensor],
+        vis_inputs: List[torch.Tensor],
         masks: Optional[torch.Tensor] = None,
         memories: Optional[torch.Tensor] = None,
     ) -> Tuple[torch.Tensor, int, int, int, int]: