Merge pull request #5389 from nouiz/printer_crash

theano/tensor/subtensor.py

@@ -967,14 +967,17 @@ class SubtensorPrinter:
         elif isinstance(r.owner.op, Subtensor):
             idxs = r.owner.op.idx_list
             inputs = list(r.owner.inputs)
-            input = inputs.pop()
+            input = inputs.pop(0)
             sidxs = []
-            inbrack_pstate = pstate.clone(precedence=-1000)
+            old_precedence = getattr(pstate, 'precedence', None)
+            try:
+                pstate.precedence = -1000
+
                 for entry in idxs:
                     if isinstance(entry, integer_types):
                         sidxs.append(str(entry))
                     elif isinstance(entry, scal.Scalar):
-                    sidxs.append(inbrack_pstate.pprinter.process(inputs.pop()))
+                        sidxs.append(pstate.pprinter.process(inputs.pop()))
                     elif isinstance(entry, slice):
                         if entry.start is None or entry.start == 0:
                             msg1 = ""
@@ -992,10 +995,15 @@ class SubtensorPrinter:
                             msg3 = ":%s" % entry.step
 
                         sidxs.append("%s:%s%s" % (msg1, msg2, msg3))
-            return "%s[%s]" % (pstate.pprinter.process(
-                input,
-                pstate.clone(precedence=1000)),
-                ", ".join(sidxs))
+            finally:
+                pstate.precedence = old_precedence
+
+            try:
+                pstate.precedence = 1000
+                sub = pstate.pprinter.process(input, pstate)
+            finally:
+                pstate.precedence = old_precedence
+            return "%s[%s]" % (sub, ", ".join(sidxs))
         else:
             raise TypeError("Can only print Subtensor.")
 

theano/tests/test_printing.py

@@ -738,3 +738,9 @@ def test_printing_scan():
                         allow_input_downcast=True)
     theano.printing.pydotprint(output, scan_graphs=True)
     theano.printing.pydotprint(f, scan_graphs=True)
+
+
+def test_subtensor():
+    x = theano.tensor.dvector()
+    y = x[1]
+    assert theano.pp(y) == "<TensorType(float64, vector)>[Constant{1}]"