lots more generics

1 files changed, 7 insertions, 6 deletions

diff --git a/pkg/gui/controllers/helpers/cherry_pick_helper.go b/pkg/gui/controllers/helpers/cherry_pick_helper.go
index e4a9b3e81..c433655d0 100644
--- a/pkg/gui/controllers/helpers/cherry_pick_helper.go
+++ b/pkg/gui/controllers/helpers/cherry_pick_helper.go
@@ -2,6 +2,7 @@ package helpers
 
 import (
 	"github.com/jesseduffield/generics/set"
+	"github.com/jesseduffield/generics/slices"
 	"github.com/jesseduffield/lazygit/pkg/commands"
 	"github.com/jesseduffield/lazygit/pkg/commands/models"
 	"github.com/jesseduffield/lazygit/pkg/gui/context"
@@ -118,12 +119,12 @@ func (self *CherryPickHelper) add(selectedCommit *models.Commit, commitsList []*
 	commitSet := self.CherryPickedCommitShaSet()
 	commitSet.Add(selectedCommit.Sha)
 
-	commitsInSet := lo.Filter(commitsList, func(commit *models.Commit, _ int) bool {
-		return commitSet.Includes(commit.Sha)
-	})
-	newCommits := lo.Map(commitsInSet, func(commit *models.Commit, _ int) *models.Commit {
-		return &models.Commit{Name: commit.Name, Sha: commit.Sha}
-	})
+	newCommits := slices.FilterThenMap(commitsList,
+		func(commit *models.Commit) bool { return commitSet.Includes(commit.Sha) },
+		func(commit *models.Commit) *models.Commit {
+			return &models.Commit{Name: commit.Name, Sha: commit.Sha}
+		},
+	)
 
 	self.getData().CherryPickedCommits = newCommits
 }