Skip to content

Commit 595fc2d

Browse files
committed
fix bug introduced in last commit
1 parent d4d2111 commit 595fc2d

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pathpy/classes/multi_order_model.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def max_order(self):
7070
"""the current maximum order of the multi-order model"""
7171
orders = list(self.layers.keys())
7272
if not orders:
73-
return -1
73+
return None
7474
else:
7575
return max(orders)
7676

@@ -129,13 +129,14 @@ def add_layers(self, max_order):
129129
"""
130130
from pathpy import ENABLE_MULTICORE_SUPPORT
131131

132+
current_max_order = self.max_order if self.max_order else -1
132133
if max_order < 0:
133134
raise PathpyError("max_order must be a positive integer not %d" % max_order)
134135

135-
if max_order <= self.max_order:
136+
if max_order <= current_max_order:
136137
Log.add("Layers up to order %d already added. Nothing changed." % self.max_order)
137138

138-
orders_to_add = list(range(self.max_order+1, max_order+1))
139+
orders_to_add = list(range(current_max_order+1, max_order+1))
139140
if len(orders_to_add) > 1 and ENABLE_MULTICORE_SUPPORT:
140141
self.__add_layers_parallel(orders_to_add)
141142
else:

0 commit comments

Comments
 (0)