import torch
from typing import Dict
from typing import Any


def get_state_dict(model: torch.nn.Module) -> Dict[str, Any]:
    """
    A proxy function that gets the state dictionary.

    Args:
        model: Torch model.
    Returns:
        State dictionary containing model weights.
    """
    return model.state_dict()