From 7a3875d0698200aaa071b42d5624457fd96b20f0 Mon Sep 17 00:00:00 2001 From: kierandidi Date: Tue, 26 Mar 2024 23:00:23 +0000 Subject: [PATCH] changed time intervall check --- proteinworkshop/datasets/pdb_dataset.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/proteinworkshop/datasets/pdb_dataset.py b/proteinworkshop/datasets/pdb_dataset.py index d569624d..0daa9d95 100644 --- 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: