dvc: some minor optimizations

This commit is contained in:
Alexander Schepanovski
2019-04-26 19:10:08 +07:00
parent c0104e2e61
commit cdba077b61
2 changed files with 10 additions and 8 deletions
+2 -2
View File
@@ -469,10 +469,10 @@ class RemoteBase(object):
def gc(self, cinfos):
from dvc.remote.local import RemoteLOCAL
used = [info[RemoteLOCAL.PARAM_CHECKSUM] for info in cinfos["local"]]
used = {info[RemoteLOCAL.PARAM_CHECKSUM] for info in cinfos["local"]}
if self.scheme != "":
used += [info[self.PARAM_CHECKSUM] for info in cinfos[self.scheme]]
used |= {info[self.PARAM_CHECKSUM] for info in cinfos[self.scheme]}
removed = False
for checksum in self.all():
+8 -6
View File
@@ -285,12 +285,14 @@ class Repo(object):
for out in stage.outs:
scheme = out.path_info["scheme"]
cache[scheme] += self._collect_used_cache(
out,
branch=branch,
remote=remote,
force=force,
jobs=jobs,
cache[scheme].extend(
self._collect_used_cache(
out,
branch=branch,
remote=remote,
force=force,
jobs=jobs,
)
)
return cache