fix

torch_geometric/nn/models/mlp.py

@@ -151,7 +151,7 @@ def forward(self, x: Tensor, return_emb: bool = False) -> Tensor:
             x = F.dropout(x, p=self.dropout, training=self.training)
             x = lin.forward(x)
 
-        return x, emb if return_emb else x
+        return (x, emb) if return_emb else x
 
     def __repr__(self) -> str:
         return f'{self.__class__.__name__}({str(self.channel_list)[1:-1]})'