fs: Accept list of paths in rm.

To comply with fsspec.

Author: daavoo

src/dvc_objects/fs/base.py

@@ -449,8 +449,13 @@ def rmdir(self, path: AnyFSPath) -> None:
     def rm_file(self, path: AnyFSPath) -> None:
         self.fs.rm_file(path)
 
-    def rm(self, path: AnyFSPath) -> None:
-        self.fs.rm(path, recursive=True)
+    def rm(
+        self,
+        path: Union[AnyFSPath, List[AnyFSPath]],
+        recursive: bool = False,
+        **kwargs,
+    ) -> None:
+        self.fs.rm(path, recursive=recursive, **kwargs)
 
     remove = rm
 

src/dvc_objects/fs/local.py

@@ -110,7 +110,10 @@ def rm_file(self, path):
         remove(path)
 
     def rm(self, path, recursive=False, maxdepth=None):
-        remove(path)
+        if isinstance(path, str):
+            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