class GraphModule(torch.nn.Module):
    def forward(self, L_x_: "f32[9][1]cpu"):
        l_x_ = L_x_

        # File: /Users/aps/Documents/ChatGPT/Job Apps/dynamo-diff/.cache/pilot-worker/model.py:8 in f, code: z = x.cos()
        z: "f32[9][1]cpu" = l_x_.cos();  l_x_ = None
        return (z,)
