Skip to content

Commit ffd50ce

Browse files
committed
Adopt FileManagerStacked to match FileManager
1 parent 650a464 commit ffd50ce

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

src/vidata/file_manager/file_manager.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -253,10 +253,15 @@ class FileManagerStacked(FileManager):
253253
...
254254
"""
255255

256-
def collect_files(self):
257-
super().collect_files()
258-
if self.files != []:
259-
files = [file.with_name(file.stem.rsplit("_", 1)[0]) for file in self.files]
256+
@staticmethod
257+
def collect_files(
258+
path: Path, file_type: str, pattern: str | None, recursive: bool = False
259+
) -> list[Path]:
260+
# def collect_files(self):
261+
files = FileManager.collect_files(path, file_type, pattern, recursive)
262+
if files != []:
263+
files = [file.with_name(file.stem.rsplit("_", 1)[0]) for file in files]
260264
files = np.unique(files)
261265

262-
self.files = natsorted(files, key=lambda p: p.name)
266+
files = natsorted(files, key=lambda p: p.name)
267+
return files

0 commit comments

Comments
 (0)