code: def forward(self, x):
      return {
          "key": {
              "emb1": self.emb1(x),
              "emb2": self.emb2(x),
          }
      }
  
imports: import pytest
  import wandb
label_desc: Use image node_id 2 for calling the EmbModel obj's underlying member method code with example usage: obj.forward(x) and returns: dict