From f4c0946f65767174d918c37cbc787794499f42e8 Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Tue, 8 Aug 2023 00:41:21 +0300 Subject: [PATCH] repo: cache: support some view options --- dvc/repo/cache.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/dvc/repo/cache.py b/dvc/repo/cache.py index 7d86b9550..3c6143319 100644 --- a/dvc/repo/cache.py +++ b/dvc/repo/cache.py @@ -1,11 +1,17 @@ import os -def check_missing(repo, rev=None): +def check_missing(repo, rev=None, max_size=None, types=None): from dvc_data.index import StorageKeyError with repo.switch(rev or "workspace"): - index = repo.index.data["repo"] + idx = repo.index.targets_view( + None, + max_size=max_size, + types=types, + ) + + index = idx.data["repo"] def onerror(_entry, _exc): pass