Fix time-zero logic

jeromekelleher · Oct 21, 2024 · 8eb75f8 · 8eb75f8
1 parent eecde47
commit 8eb75f8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/sc2ts/info.py b/sc2ts/info.py
@@ -492,10 +492,8 @@ def _preprocess_nodes(self, show_progress):
         self.nodes_metadata = {}
         self.nodes_sample_group = collections.defaultdict(list)
         samples = ts.samples()
-        last_sample = ts.node(samples[-1])
 
-        self.nodes_date[last_sample.id] = last_sample.metadata["date"]
-        self.time_zero_as_date = self.nodes_date[last_sample.id]
+        self.time_zero_as_date = np.array([self.date], dtype="datetime64[D]")[0]
         self.earliest_pango_lineage = {}
         self.pango_lineage_samples = collections.defaultdict(list)