Function deprecated_graph_mode_only

tensorflow/python/framework/test_util.py:1178–1224

Execute the decorated test in graph mode. This function returns a decorator intended to be applied to tests that are not compatible with eager mode. When this decorator is applied, the test body will be run in an environment where API calls construct graphs instead of executing eagerly.

(func=None)

def deprecated_graph_mode_only(func=None):
  """Execute the decorated test in graph mode.

  This function returns a decorator intended to be applied to tests that are not
  compatible with eager mode. When this decorator is applied, the test body will
  be run in an environment where API calls construct graphs instead of executing
  eagerly.

  `deprecated_graph_mode_only`, `run_v1_only`, `run_v2_only`, and
  `run_in_graph_and_eager_modes` are available decorators for different
  v1/v2/eager/graph combinations.

  Args:
    func: function to be annotated. If `func` is None, this method returns a
      decorator the can be applied to a function. If `func` is not None this
      returns the decorator applied to `func`.

  Returns:
    Returns a decorator that will run the decorated test method in graph mode.
  """

  def decorator(f):
    if tf_inspect.isclass(f):
      setup = f.__dict__.get("setUp")
      if setup is not None:
        setattr(f, "setUp", decorator(setup))

      for name, value in f.__dict__.copy().items():
        if (callable(value) and
            name.startswith(unittest.TestLoader.testMethodPrefix)):
          setattr(f, name, decorator(value))

      return f

    def decorated(self, *args, **kwargs):
      if tf2.enabled():
        with context.graph_mode():
          return f(self, *args, **kwargs)
      else:
        return f(self, *args, **kwargs)

    return decorated

  if func is not None:
    return decorator(func)

  return decorator


run_deprecated_v1 = deprecated_graph_mode_only