fixed problem

theano/sandbox/multinomial.py

@@ -58,6 +58,11 @@ class MultinomialFromUniform(Op):
 #        return (6,)
 
     def c_code(self, node, name, ins, outs, sub):
+        # support old pickled graphs
+        if len(ins) == 2:
+            (pvals, unis) = ins
+            n = 1
+        else:
             (pvals, unis, n) = ins
         (z,) = outs
         if self.odtype == 'auto':
@@ -269,13 +274,7 @@ class GpuMultinomialFromUniform(MultinomialFromUniform, GpuOp):
         """ % locals()
 
     def c_code(self, node, name, ins, outs, sub):
-        # support old pickled graphs
-        if len(ins) == 2:
         (pvals, unis) = ins
-            n_samples = 1
-        else:
-            (pvals, unis, n_samples) = ins
-
         (z,) = outs
 
         fail = sub['fail']