code: def _compute_rdp_tree_restart(sigma, steps_list, alpha):
      """Computes RDP of the Tree Aggregation Protocol at order alpha."""
      if np.isinf(alpha):
          return np.inf
      tree_depths = [
          math.floor(math.log2(float(steps))) + 1
          for steps in steps_list
          if steps > 0
      ]
      return _compute_gaussian_rdp(
          alpha=alpha,
          sum_sensitivity_square=sum(tree_depths),
          sigma=sigma,
      )
  
file_path: privacy/tensorflow_privacy/privacy/analysis/tree_aggregation_accountant.py
imports: import functools
  import math
  from typing import Collection, Optional, Union
  import dp_accounting
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: _compute_rdp_tree_restart(sigma, steps_list, alpha) and returns: _compute_gaussian_rdp, np