Merge pull request #1003 from fvisin/fix_recursion_limit

Fix Blocks ignores python's recursionlimit setting
mila-iqia · Feb 24, 2016 · 5107756 · 5107756
2 parents 84f1b6d + eb1e242
commit 5107756
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/blocks/utils/__init__.py b/blocks/utils/__init__.py
@@ -483,7 +483,8 @@ def print_shape(x, header=None):
 def change_recursion_limit(limit):
     """Temporarily changes the recursion limit."""
     old_limit = sys.getrecursionlimit()
-    sys.setrecursionlimit(limit)
+    if old_limit < limit:
+        sys.setrecursionlimit(limit)
     yield
     sys.setrecursionlimit(old_limit)