tff.test.with_context

Stay organized with collections Save and categorize content based on your preferences.

Returns a decorator for running a test in a tff.Context.

context_fn A Callable that constructs a tff.Context to install beore invoking the decorated function.
environment_fn A Callable that constructs a list of contextlib.AbstractContextManager to enter before invoking the decorated function.