From 4afae0fede8928e2709368e3ba4665b8b883ad58 Mon Sep 17 00:00:00 2001
From: Moritz Hoffmann <antiguru@gmail.com>
Date: Mon, 19 Feb 2024 10:08:52 -0500
Subject: [PATCH] Support container GATs

Signed-off-by: Moritz Hoffmann <antiguru@gmail.com>
---
 Cargo.toml        | 3 ++-
 src/collection.rs | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 528913e33..17af8e0a6 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -50,7 +50,8 @@ timely = {workspace = true}
 
 [workspace.dependencies]
 #timely = { version = "0.12", default-features = false }
-timely = { git = "https://github.com/TimelyDataflow/timely-dataflow", default-features = false }
+#timely = { git = "https://github.com/TimelyDataflow/timely-dataflow", default-features = false }
+timely = { git = "https://github.com/antiguru/timely-dataflow", branch = "container_gat", default-features = false }
 #timely = { path = "../timely-dataflow/timely/", default-features = false }
 
 [features]
diff --git a/src/collection.rs b/src/collection.rs
index f2daf8ad4..7ac5dc4fb 100644
--- a/src/collection.rs
+++ b/src/collection.rs
@@ -416,10 +416,10 @@ impl<G: Scope, D: Data, R: Semigroup> Collection<G, D, R> where G::Timestamp: Da
     ///          .inspect_batch(|t,xs| println!("errors @ {:?}: {:?}", t, xs));
     /// });
     /// ```
-    pub fn inspect_batch<F>(&self, func: F) -> Collection<G, D, R>
+    pub fn inspect_batch<F>(&self, mut func: F) -> Collection<G, D, R>
     where F: FnMut(&G::Timestamp, &[(D, G::Timestamp, R)])+'static {
         self.inner
-            .inspect_batch(func)
+            .inspect_batch(move |ts, b| func(ts, &*b))
             .as_collection()
     }
     /// Attaches a timely dataflow probe to the output of a Collection.