GitLab

* Removed usage of `tff.test.TestCase` and `tff.test.main()` from `aggregators` package.
* Cleaned up test parameter what are unused.

PiperOrigin-RevId: 449080424

PiperOrigin-RevId: 449053292

* Removed usage of `tff.test.TestCase` and `tff.test.main()` from `learning` package.
* Cleaned up test parameter what are unused.

PiperOrigin-RevId: 449044021

This change removes the need to alias the module when it is imported and makes it easier to reason about which test utilities we are using in the module. Additionally, this change applies this pattern more consistently across TFF.

PiperOrigin-RevId: 449005182

Internally it's easier to use the functions than the class, because the functions can be broken up and owned by the package with the appropriate visibility. The class acts as an object where it needs visibility into every other package and ever other package needs visibility to it. This creates a circular dependencies.

Additionally, it's more convenient for external usage of an API to resemble internal usage of the same API, so that upstreaming research code is easier.

PiperOrigin-RevId: 448995317

This change removes the need to alias the module when it is imported and makes it easier to reason about which test utilities we are using in the module.

Additionally, this change applies this pattern more consistently across TFF.

PiperOrigin-RevId: 448985467

PiperOrigin-RevId: 448983692

PiperOrigin-RevId: 448978762

* Removed dependency.
* Created `computation_test_utils` to contain utilities for testing computation protos.

PiperOrigin-RevId: 448966416

Add asyncio.run call to metrics manager release calls to ensure compatibility with a98b5ed6.

PiperOrigin-RevId: 448588151

This removes the need to alias the module when it is imported and makes it easier to reason about which test utilities we are using in the module.

PiperOrigin-RevId: 448584405

This removes the need to alias the module when it is imported and makes it easier to reason about which test utilities we are using in the module.

PiperOrigin-RevId: 448584190

PiperOrigin-RevId: 448517906

container as the request type.

This is a necessary step to enabling auto-control-dependencies (ACD) in the
`tensorflow_wrapper`. ACD is not compatible with `tff.structure.Struct`.

PiperOrigin-RevId: 448517464

This is a necessary step to enabling auto-control-dependencies (ACD) in the
`tensorflow_wrapper`. ACD is not compatible with `tff.structure.Struct`.

PiperOrigin-RevId: 448509972

PiperOrigin-RevId: 448506935

This function is only being used in one place, its easy to use the `is_assignable_from` function on the type itself.

PiperOrigin-RevId: 448367278

This function is only being used in one place.

PiperOrigin-RevId: 448336618

Replace `assert_type_string` with the function it is forward declaring, which is more readable at the call site.

PiperOrigin-RevId: 448336108

PiperOrigin-RevId: 448309382

container used in the incoming type, instead of `tff.structure.Struct`.

This is a necessary step to enabling auto-control-dependencies (ACD) in the
`tensorflow_wrapper`. ACD is not compatible with `tff.structure.Struct`.

PiperOrigin-RevId: 448303704

return `tff.structure.Struct`.

This is a necessary step to enabling auto-control-dependencies (ACD) in the
`tensorflow_wrapper`. ACD is not compatible with `tff.structure.Struct`.

PiperOrigin-RevId: 448297886

* Moved `computation_base' to the `computation` package.
* Moved `compiler_pipeline` to the `execution_contexts` package. The `compiler` package should not depend on the `computation` package; however, the `compiler_pipeline` is really just a helper to the `execution_contexts` package so it's better placed there anyway.
* Added the `execution_contexts` package to the `computation` package visibility.

PiperOrigin-RevId: 448027843

Embedding concatenation logic into `EliasGammaEncodedSumFactory` so that it can encode any structure.

PiperOrigin-RevId: 447808142

Create a mirror of `tff.simulation.compose_dataset_computation_with_iterative_process` for `tff.learning.templates.LearningProcess`.

This functionality also preserves non-`next` properties.

Finally, this change relaxes the prior restriction on `LearningProcess` that its second argument to `next` must be a sequence type.

PiperOrigin-RevId: 447771902

PiperOrigin-RevId: 447078420

It's not clear that there is a quick way to resolve this warning that doesn't make the code more confusing to read.

PiperOrigin-RevId: 446721461

PiperOrigin-RevId: 446598740

PiperOrigin-RevId: 446584986

PiperOrigin-RevId: 446576253

PiperOrigin-RevId: 446557729

The two functions `robust_aggregator` and `compression_aggregator` now use a param `debug_measurements_fn` to get the function for adding debug measurements. The param `add_debug_measurements` is deprecated.

PiperOrigin-RevId: 446527276

This code is also just a bit harder to read as a comprehension.

PiperOrigin-RevId: 446510425

PiperOrigin-RevId: 446506249

Replaced usage of deprecated `abc.abstractproperty` with using both `property` and `abc.abstractmethod`. See https://docs.python.org/3/library/abc.html#abc.abstractproperty for more information.

PiperOrigin-RevId: 446501681

PiperOrigin-RevId: 446480498

PiperOrigin-RevId: 446468087

Convert argument type into tff.Type before validating in the constructor of CardinalityFreeDataDescriptor.

PiperOrigin-RevId: 446235480

Move the `tf.errors.PermissionDeniedError` exception handling so that it captures failures to close the file handle.

When the file handle is closed as a result of an error, this change correctly returns the `FileReleaseManagerPermissionDeniedError`, "Could not append a value to the file '...'. It is possible that this file is compressed or encoded. Please use write mode instead of append mode to release values to this file using a `tff.program.CSVFileReleaseManager`."

PiperOrigin-RevId: 446203127

PiperOrigin-RevId: 446005411