Fixing merge lbm seq

This commit is contained in:
Louis 2026-02-24 11:30:54 +01:00
parent 88024bd343
commit e6c38905b3

View file

@ -3,18 +3,22 @@ library(stringr)
args <- commandArgs(trailingOnly = TRUE)
path <- args[1]
mode <- args[2]
base_folder <- str_remove(string = path, pattern = "/tmp[0-9]*$")
print(base_folder)
epoch <- str_extract(string = path, pattern = "(?<=tmp)([0-9]*)")
flist <- list.files(path, pattern = paste0(mode, ".Rds"), full.names = TRUE)
for (mode in c("seq", "para", "notrans")) {
flist <- list.files(path, pattern = paste0(mode, ".Rds"), full.names = TRUE)
if (length(flist) == 0) {
warning("No files corresponding to ", mode, " found.")
next
}
merged_res <- lapply(flist, readRDS)
stopifnot("No files corresponding to mode found." = length(flist) > 0)
names(merged_res) <- paste0("Rank", seq_along(flist) + 1)
merged_res <- lapply(flist, readRDS)
names(merged_res) <- paste0("Rank", seq_along(flist) + 1)
saveRDS(merged_res, file.path(base_folder, paste0(mode, "_", epoch, ".Rds")))
saveRDS(merged_res, file.path(base_folder, paste0(mode, "_", epoch, ".Rds")))
}