changed time intervall check

a-r-j · Mar 26, 2024 · 7a3875d · 7a3875d
1 parent 878fa96
commit 7a3875d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/proteinworkshop/datasets/pdb_dataset.py b/proteinworkshop/datasets/pdb_dataset.py
@@ -55,9 +55,7 @@ def __init__(
         self.split_type = split_type
         self.split_sequence_similarity = split_sequence_similiarity
         self.overwrite_sequence_clusters = overwrite_sequence_clusters
-        if split_time_frames is None:
-            self.split_time_frames = split_time_frames
-        else:
+        if self.split_type == "time_cutoff":
             try:
                 self.split_time_frames = [np.datetime64(date) for date in split_time_frames]
             except: