Skip to content

Commit 467cbfc

Browse files
committed
Adding delete_mode function that was missing from mock assignment
1 parent d056009 commit 467cbfc

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

Scripts/assignment/emme_bindings/mock_project.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -564,6 +564,13 @@ def create_mode(self, mode_type: str, idx: str) -> 'Mode':
564564
mode = Mode(idx, mode_type)
565565
self._modes[idx] = mode
566566
return mode
567+
568+
def delete_mode(self, idx: str, cascade: bool = False):
569+
mode = self._modes[idx]
570+
if cascade:
571+
for link in self._links:
572+
self._links[link].modes -= {mode}
573+
self._modes.pop(idx)
567574

568575
def node(self, idx: int) -> 'Node':
569576
idx = int(idx)

0 commit comments

Comments
 (0)