diff --git a/src/Model/Friction/Friction.py b/src/Model/Friction/Friction.py index d59854d6..1aa46b55 100644 --- a/src/Model/Friction/Friction.py +++ b/src/Model/Friction/Friction.py @@ -167,25 +167,26 @@ class Friction(SQLSubModel): deleted = (next(it) == 1) begin_rk = next(it) end_rk = next(it) - begin_strickler_pid = next(it) - end_strickler_pid = next(it) + begin_strickler_pid = int(next(it)) + end_strickler_pid = int(next(it)) owner_scenario = next(it) # Get stricklers begin_strickler = None - if int(begin_strickler_pid) != -1: + if begin_strickler_pid != -1: begin_strickler = next( filter( - lambda s: s.pamhyr_id == begin_strickler_pid, + lambda s: s.id == begin_strickler_pid, stricklers ) ) - begin_strickler = None - if int(end_strickler_pid) != -1: - begin_strickler = next( + end_strickler = None + if end_strickler_pid != -1: + end_strickler = next( filter( - lambda s: s.id == begin_strickler_pid, stricklers + lambda s: s.id == begin_strickler_pid, + stricklers ) )