code: def _partition_metrics_private_partitions(
      prob_keep: float,
  ) -> metrics.PartitionsInfo:
      kept_partitions = metrics.MeanVariance(
          mean=prob_keep, var=prob_keep * (1 - prob_keep)
      )
      return metrics.PartitionsInfo(
          public_partitions=False,
          num_dataset_partitions=1,
          kept_partitions=kept_partitions,
      )
  
imports: import copy
  import pipeline_dp
  from analysis import metrics
  import dataclasses
  from typing import List, Tuple, Callable
  import math
label_desc: Use image node_id 6 for calling a global function with example usage: _partition_metrics_private_partitions(prob_keep) and returns: metrics