dvc: some minor optimizations

mathemaphysics · Apr 26, 2019 · cdba077 · cdba077
1 parent c0104e2
commit cdba077
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/dvc/remote/base.py b/dvc/remote/base.py
@@ -469,10 +469,10 @@ def all(self):
     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():

diff --git a/dvc/repo/__init__.py b/dvc/repo/__init__.py
@@ -285,12 +285,14 @@ def used_cache(
 
                 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