tff.test.assert_contains_secure_aggregation

View source on GitHub

Asserts that comp contains at least one secure aggregation call.

comp A tff.Computation, often a function annotated with tff.federated_computation or tff.tf_computation. Note that polymorphic functions (those without the types of their arguments explicitly specified) will not yet be tff.Computations.

AssertionError if comp does not contain a secure aggregation call. ValueError if comp contains a call whose target function cannot be identified. This may result from calls to references or other indirect structures.