From a94fdc3aebdcf3f9ebc3d7a5fa8799fa9618262e Mon Sep 17 00:00:00 2001 From: David Rotermund Date: Sat, 10 Aug 2024 20:55:37 +0200 Subject: [PATCH] Add files via upload --- functions/get_trials.py | 1 + 1 file changed, 1 insertion(+) diff --git a/functions/get_trials.py b/functions/get_trials.py index 8c687d9..abe33d2 100644 --- a/functions/get_trials.py +++ b/functions/get_trials.py @@ -14,5 +14,6 @@ def get_trials(path: str, experiment_id: int) -> torch.Tensor: list_int: list[int] = [] for i in range(0, len(list_str)): list_int.append(int(list_str[i].split("_Trial")[-1].split("_Part")[0])) + list_int = sorted(list_int) return torch.tensor(list_int).unique()