code: def test_mlp_return_emb():
      x = torch.randn(4, 16)
  
      mlp = MLP([16, 32, 1])
  
      out, emb = mlp(x, return_emb=True)
      assert out.size() == (4, 1)
      assert emb.size() == (4, 32)
  
      out, emb = mlp(x, return_emb=False)
      assert out.size() == (4, 1)
      assert emb is None
  
imports: import pytest
  import torch
  from torch_geometric.nn import MLP
label_desc: Use image node_id 3 for calling a global function with example usage: test_mlp_return_emb() without return types