From 7a65d67e7fa772773a04d06149e000495832ef7c Mon Sep 17 00:00:00 2001
From: Peter Rowlands <peter@pmrowla.com>
Date: Thu, 8 Dec 2022 13:01:36 +0900
Subject: [PATCH] worktree: support recursive/with_deps targets

---
 dvc/repo/worktree.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/dvc/repo/worktree.py b/dvc/repo/worktree.py
index 948358d86..cf5af4850 100644
--- a/dvc/repo/worktree.py
+++ b/dvc/repo/worktree.py
@@ -71,11 +71,14 @@ def worktree_view(
 
 
 def fetch_worktree(
-    repo: "Repo", remote: "Remote", targets: Optional["TargetType"] = None
+    repo: "Repo",
+    remote: "Remote",
+    targets: Optional["TargetType"] = None,
+    **kwargs,
 ) -> int:
     from dvc_data.index import save
 
-    view = worktree_view(repo.index, targets=targets)
+    view = worktree_view(repo.index, targets=targets, **kwargs)
     index = view.data["repo"]
     for key, entry in index.iteritems():
         entry.fs = remote.fs
@@ -92,12 +95,19 @@ def fetch_worktree(
 
 
 def push_worktree(
-    repo: "Repo", remote: "Remote", targets: Optional["TargetType"] = None
+    repo: "Repo",
+    remote: "Remote",
+    targets: Optional["TargetType"] = None,
+    **kwargs,
 ) -> int:
     from dvc_data.index import checkout
 
     view = worktree_view(
-        repo.index, push=True, targets=targets, latest_only=remote.worktree
+        repo.index,
+        push=True,
+        targets=targets,
+        latest_only=remote.worktree,
+        **kwargs,
     )
     new_index = view.data["repo"]
     if remote.worktree: