def METHOD_NAME(self):
    class SimpleTest(torch.nn.Module):
        def forward(self, x):
            return torch.relu(x + 3.0)
    st = SimpleTest()
    traced = symbolic_trace(st)
    f = BytesIO()
    with PackageExporter(f) as pe:
        pe.save_pickle("model", "model.pkl", traced)
    f.seek(0)
    pi = PackageImporter(f)
    loaded_traced = pi.load_pickle("model", "model.pkl")
    input = torch.rand(2, 3)
    self.assertEqual(loaded_traced(input), traced(input))