Use "auto" with GENERATOR instead of std::function

[folly.git] / folly / gen / Base.h

diff --git a/folly/gen/Base.h b/folly/gen/Base.h
index a550393554a6fa490b633753886b1b417c03ee97..dd74a092a68046adc2e889839034ed82d0cab914 100644 (file)
--- a/folly/gen/Base.h
+++ b/folly/gen/Base.h
@@ -503,9 +503,8 @@ Yield generator(Source&& source) {
  *
  *  auto gen = GENERATOR(int) { yield(1); yield(2); };
  */
-#define GENERATOR(TYPE)                            \
-  ::folly::gen::detail::GeneratorBuilder<TYPE>() + \
-   [=](const std::function<void(TYPE)>& yield)
+#define GENERATOR(TYPE) \
+  ::folly::gen::detail::GeneratorBuilder<TYPE>() + [=](auto&& yield)
 
 /*
  * empty() - for producing empty sequences.