From 4ace91c5f12b6d6706a05dcbaa44e8cadc36bcc6 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 19 Jan 2021 14:15:16 +0100
Subject: fix off-by-one in graph-based drift compensation

the bug manifested whenever a "transition end" node was chosen
---
 lib/lennart/DataProcessor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'lib')

diff --git a/lib/lennart/DataProcessor.py b/lib/lennart/DataProcessor.py
index 2b495f6..c1319c9 100644
--- a/lib/lennart/DataProcessor.py
+++ b/lib/lennart/DataProcessor.py
@@ -300,10 +300,10 @@ class DataProcessor:
 
         prev_nodes = [0]
         prev_drifts = [0]
+        node_drifts = [0]
         edge_srcs = list()
         edge_dsts = list()
         csr_weights = list()
-        node_drifts = list()
 
         # (transition index) -> [candidate 0/start node, candidate 0/end node, candidate 1/start node, ...]
         nodes_by_transition_index = dict()
-- 
cgit v1.2.3