From 82e7b224431d05f3d77eee527b36f61862ba80b3 Mon Sep 17 00:00:00 2001
From: Anders Olsson <anders.e.olsson@gmail.com>
Date: Mon, 13 Jun 2022 11:37:56 +0200
Subject: [PATCH] Use flatten instead of flat_map

---
 src/batch.rs   | 4 ++--
 src/history.rs | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/batch.rs b/src/batch.rs
index 4444047..34e5cc4 100644
--- a/src/batch.rs
+++ b/src/batch.rs
@@ -127,8 +127,8 @@ impl Batch {
     ) {
         let this_agent = composition
             .iter()
-            .flat_map(|teams| teams.iter())
-            .flat_map(|team| team.iter())
+            .flatten()
+            .flatten()
             .cloned()
             .collect::<HashSet<_>>();
 
diff --git a/src/history.rs b/src/history.rs
index 1cb4ef6..ea6acab 100644
--- a/src/history.rs
+++ b/src/history.rs
@@ -27,8 +27,8 @@ impl History {
     ) -> Self {
         let this_agent = composition
             .iter()
-            .flat_map(|teams| teams.iter())
-            .flat_map(|team| team.iter())
+            .flatten()
+            .flatten()
             .cloned()
             .collect::<HashSet<_>>();