From 1a2c91eb5ae6f74ea7d760c702f87f6b150af14e Mon Sep 17 00:00:00 2001 From: Saugat Pachhai Date: Mon, 23 Nov 2020 12:44:43 +0545 Subject: [PATCH] Use PathStringTrie consistently (#4941) This one was missed for some reason. --- dvc/tree/repo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dvc/tree/repo.py b/dvc/tree/repo.py index 89f9ea682..e696013e5 100644 --- a/dvc/tree/repo.py +++ b/dvc/tree/repo.py @@ -7,12 +7,12 @@ from itertools import takewhile from typing import TYPE_CHECKING, Callable, Optional, Tuple, Union from funcy import lfilter, wrap_with -from pygtrie import StringTrie from dvc.dvcfile import is_valid_filename from dvc.hash_info import HashInfo from dvc.path_info import PathInfo from dvc.utils import file_md5, is_exec +from dvc.utils.collections import PathStringTrie from dvc.utils.fs import copy_fobj_to_file, makedirs from ._metadata import Metadata @@ -62,7 +62,7 @@ class RepoTree(BaseTree): # pylint:disable=abstract-method self.root_dir = repo.root_dir self._traverse_subrepos = subrepos - self._subrepos_trie = StringTrie(separator=os.sep) + self._subrepos_trie = PathStringTrie() """Keeps track of each and every path with the corresponding repo.""" self._subrepos_trie[self.root_dir] = repo