def METHOD_NAME(self) -> Dict[str, int]:
    """Get shape of FederatedDatasets by node ids.
    Returns:
        Includes [`sample_sizes`][fedbiomed.researcher.datasets.FederatedDataSet.sample_sizes] by node_ids.
    """
    shapes_dict = {}
    for node_id, node_data_size in zip(self.node_ids(),
                                       self.sample_sizes()):
        shapes_dict[node_id] = node_data_size
    return shapes_dict