def METHOD_NAME():
    """Returns a tuple of (old_layers, new_layers, and optimizer)."""
    model = SimpleModel(num_features=1, num_classes=10)
    policy: Mapping[Type[torch.nn.Module], module_surgery.ReplacementFunction] = {
        torch.nn.Linear: _CopyLinear.from_linear
    }
    opt = torch.optim.SGD(model.parameters(), lr=.001)
    orig_linear_modules = [model.fc1, model.fc2]
    module_surgery.replace_module_classes(model, policies=policy, optimizers=opt)
    new_linear_modules = [model.fc1, model.fc2]
    return orig_linear_modules, new_linear_modules, opt