fs: Accept list of paths in rm.

To comply with fsspec.

Author: daavoo

src/dvc_objects/fs/base.py

@@ -378,7 +378,7 @@ def rmdir(self, path: AnyFSPath) -> None:
     def rm_file(self, path: AnyFSPath) -> None:
         self.fs.rm_file(path)
 
-    def rm(self, path: AnyFSPath) -> None:
+    def rm(self, path: Union[AnyFSPath, List[AnyFSPath]]) -> None:
         self.fs.rm(path, recursive=True)
 
     remove = rm

src/dvc_objects/fs/local.py

@@ -113,7 +113,10 @@ def rm_file(self, path):
         remove(path)
 
     def rm(self, path, recursive=False, maxdepth=None):
-        remove(path)
+        if not isinstance(path, list):
+            path = [path]
+        for p in path:
+            remove(p)
 
     def cp_file(self, path1, path2, **kwargs):
         return self.copy(path1, path2, **kwargs)

tests/fs/test_localfs.py

@@ -50,6 +50,16 @@ def test_local_fs_isfile(tmp_path):
     assert not fs.isfile(fspath(tmp_path / "not-existing-file"))
 
 
+def test_local_fs_rm(tmp_path):
+    (tmp_path / "file").write_text("file", encoding="utf8")
+    (tmp_path / "file2").write_text("file2", encoding="utf8")
+
+    fs = LocalFileSystem()
+    fs.remove([tmp_path / "file", tmp_path / "file2"])
+    assert not fs.exists(tmp_path / "file")
+    assert not fs.exists(tmp_path / "file2")
+
+
 def convert_to_sets(walk_results):
     return [
         (root, set(dirs), set(nondirs)) for root, dirs, nondirs in walk_results