diff --git a/src/exo/stdlib/scheduling.py b/src/exo/stdlib/scheduling.py index 6241380f..b596725a 100644 --- a/src/exo/stdlib/scheduling.py +++ b/src/exo/stdlib/scheduling.py @@ -270,7 +270,7 @@ def replace_all(proc, subprocs, mem_aware=True): if "failed to find matches" in str(e): break raise - except (_UnificationError, MemoryError): + except (_UnificationError, MemoryError, NotImplementedError): i += 1 return proc