def METHOD_NAME(
    fed_dir: Path, num_classes: int = 10
) -> Callable[[str], RayClient]:
    """Function that loads a Ray (Virtual) Client.
    Args:
        fed_dir (Path): Path containing local datasets in the form ./client_id/train.pt
        num_classes (int, optional): Number of classes. Defaults to 10.
    Returns:
        Callable[[str], RayClient]: [description]
    """
    def client_fn(cid: str) -> RayClient:
        # create a single client instance
        return RayClient(cid=cid, fed_dir=fed_dir, num_classes=num_classes)
    return client_fn