code: def _make_quantile_estimator_query(
      initial_estimate,
      target_quantile,
      learning_rate,
      below_estimate_stddev,
      expected_num_records,
      geometric_update,
      tree_aggregation=False,
  ):
      if expected_num_records is not None:
          if tree_aggregation:
              return (
                  quantile_estimator_query.TreeQuantileEstimatorQuery(
                      initial_estimate,
                      target_quantile,
                      learning_rate,
                      below_estimate_stddev,
                      expected_num_records,
                      geometric_update,
                  )
              )
          else:
              return quantile_estimator_query.QuantileEstimatorQuery(
                  initial_estimate,
                  target_quantile,
                  learning_rate,
                  below_estimate_stddev,
                  expected_num_records,
                  geometric_update,
              )
      else:
          if tree_aggregation:
              raise ValueError(
                  "Cannot set expected_num_records to None for tree aggregation."
              )
          return (
              quantile_estimator_query.NoPrivacyQuantileEstimatorQuery(
                  initial_estimate,
                  target_quantile,
                  learning_rate,
                  geometric_update,
              )
          )
  
file_path: privacy/tensorflow_privacy/privacy/dp_query/quantile_estimator_query_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_privacy.privacy.dp_query import quantile_estimator_query
  from tensorflow_privacy.privacy.dp_query import test_utils
label_desc: Use image node_id 1 for calling a global function with example usage: _make_quantile_estimator_query(initial_estimate, target_quantile, learning_rate, below_estimate_stddev, expected_num_records, geometric_update, tree_aggregation) and returns: quantile_estimator_query, quantile_estimator_query, quantile_estimator_query