EXPECT_EQ(NumCheckpointShardsWritten().value(), 10);
}

TEST(MetricsTest, TestShardingCallbackDescription) {
  EXPECT_EQ(ShardingCallbackDescription().value(), "");
  ShardingCallbackDescription().Set("foo");
  EXPECT_EQ(ShardingCallbackDescription().value(), "foo");
}

}  // namespace metrics

// Returns "/tensorflow/core/checkpoint/sharding/callback_description" cell
// which describes the callback used to shard the checkpoint during saving.
monitoring::GaugeCell<std::string>& ShardingCallbackDescription();

}  // namespace metrics
}  // namespace tensorflow

  return *sharding_callback_duration->GetCell();
}

monitoring::CounterCell& NumCheckpointShardsWritten() {
  return *num_checkpoint_shards_written->GetCell();
}

monitoring::GaugeCell<std::string>& ShardingCallbackDescription() {
  return *sharding_callback_description->GetCell();
}

}  // namespace metrics