reset based on correct flag

molmod · Dec 5, 2024 · 0e5f68c · 0e5f68c
1 parent 75158ad
commit 0e5f68c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psiflow/learning.py b/psiflow/learning.py
@@ -86,7 +86,7 @@ def evaluate_outputs(
             isnan(errors[i]),
         )
 
-        _ = assign_identifier(state, identifier, error_discard)
+        _ = assign_identifier(state, identifier, reset)
         assigned = unpack_i(_, 0)
         identifier = unpack_i(_, 1)
         processed_states.append(assigned)