changed time intervall check

Author: kierandidi

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: