from tensorflow.contrib.framework import deprecated_arg_values

class ExportMonitor(EveryN):
  """Monitor that exports Estimator every N steps."""

  # TODO(philstahlfeld): Investigate switching export.export_estimator
  # configuration values to **kwargs so that updates to the export_estimator
  # function don't have to be reflected here.
  @deprecated_arg_values(
      "2016-09-23",
      "The signature of the input_fn accepted by export is changing to be "
      "consistent with what's used by tf.Learn Estimator's train/evaluate. "
      "input_fn (and in most cases, input_feature_key) will both become "
      "required args.",
