提交 332b0b85 编辑于 作者: Galen Andrew's avatar Galen Andrew 提交者: tensorflow-copybara
浏览文件

Add DpEvent to return value of get_noised_result. For most DPQueries, the...

Add DpEvent to return value of get_noised_result. For most DPQueries, the default UnsupportedDpEvent is returned, pending further development.

PiperOrigin-RevId: 394137614
上级 0933fbed
......@@ -63,6 +63,7 @@ py_test(
srcs = ["differential_privacy_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"], # TODO(b/198312839): Replace kokoro tests after TFP Pip release
deps = [
":differential_privacy",
":factory",
......@@ -245,6 +246,7 @@ py_test(
srcs = ["quantile_estimation_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"], # TODO(b/198312839): Replace kokoro tests after TFP Pip release
deps = [
":quantile_estimation",
"//tensorflow_federated/python/core/api:test_case",
......
......@@ -349,7 +349,7 @@ class DifferentiallyPrivateFactory(factory.UnweightedAggregationFactory):
(new_agg_state, agg_result,
agg_measurements) = record_agg_process.next(agg_state, record)
result, new_query_state = intrinsics.federated_map(
result, new_query_state, _ = intrinsics.federated_map(
get_noised_result, (agg_result, query_state))
query_metrics = intrinsics.federated_map(derive_metrics, new_query_state)
......
......@@ -148,7 +148,7 @@ class PrivateQuantileEstimationProcess(estimation_process.EstimationProcess):
# like basic sum, so we won't surface its measurements.
del agg_measurements
_, new_quantile_query_state = intrinsics.federated_map(
_, new_quantile_query_state, _ = intrinsics.federated_map(
get_noised_result, (agg_result, quantile_query_state))
return intrinsics.federated_zip((new_quantile_query_state, new_agg_state))
......
......@@ -22,6 +22,7 @@ py_test(
srcs = ["hierarchical_histogram_factory_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"],
deps = [
":build_tree_from_leaf",
":hierarchical_histogram_factory",
......@@ -53,6 +54,7 @@ py_test(
srcs = ["hierarchical_histogram_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"],
deps = [
":hierarchical_histogram",
"//tensorflow_federated/python/core/api:test_case",
......
......@@ -78,6 +78,7 @@ py_cpu_gpu_test(
python_version = "PY3",
shard_count = 10,
srcs_version = "PY3",
tags = ["nokokoro"], # TODO(b/198312839): Replace kokoro tests after TFP Pip release
deps = [
":client_weight_lib",
":federated_averaging",
......@@ -156,6 +157,7 @@ py_cpu_gpu_test(
srcs = ["federated_sgd_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"], # TODO(b/198312839): Replace kokoro tests after TFP Pip release
deps = [
":client_weight_lib",
":federated_sgd",
......@@ -304,6 +306,7 @@ py_test(
srcs = ["model_update_aggregator_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"], # TODO(b/198312839): Replace kokoro tests after TFP Pip release
deps = [
":model_update_aggregator",
"//tensorflow_federated/python/aggregators:factory",
......
......@@ -112,6 +112,7 @@ py_test(
srcs = ["training_process_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"],
deps = [
":keras_utils",
":model",
......
......@@ -56,6 +56,7 @@ py_test(
srcs = ["federated_averaging_e2e_test.py"],
python_version = "PY3",
srcs_version = "PY3",
tags = ["nokokoro"], # TODO(b/198312839): Replace kokoro tests after TFP Pip release
deps = [
":learning_test_models",
"//tensorflow_federated",
......
支持 Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册