code: def test_watch_graph_torch_jit(mock_run, capsys):
      run = mock_run(use_magic_mock=True)
  
      class Net(nn.Module):
          def __init__(self):
              super().__init__()
              self.layer_1 = nn.Linear(10, 2)
  
          def forward(self, x):
              return self.layer_1(x)
  
      net = torch.jit.script(Net())
      run.watch(net, log_graph=True)
  
      outerr = capsys.readouterr()
      assert "skipping graph tracking" in outerr.err
  
file_path: wandb/tests/pytest_tests/unit_tests/test_torch.py
imports: import pytest
  import torch
  import torch.nn
  import wandb
label_desc: Use image node_id 6 for calling a global function with example usage: test_watch_graph_torch_jit(mock_run, capsys) and returns: self