correct import paths

theano/tensor/tests/_test_mpi_roundtrip.py

@@ -6,7 +6,7 @@ from mpi4py import MPI
 comm = MPI.COMM_WORLD
 import theano
 from theano.tensor.io import send, recv, mpi_cmp
-from theano.gof.graph import sort_schedule_fn
+from theano.gof.sched import sort_schedule_fn
 import numpy as np
 from sys import stdout
 

theano/tensor/tests/test_mpi.py

@@ -2,7 +2,7 @@ from theano.tensor.io import send, recv, mpi_cmp, MPISend, MPISendWait
 import theano
 import subprocess
 import os
-from theano.gof.graph import sort_schedule_fn
+from theano.gof.sched import sort_schedule_fn
 mpi_scheduler = sort_schedule_fn(mpi_cmp)
 mpi_linker = theano.OpWiseCLinker(schedule=mpi_scheduler)
 mpi_mode = theano.Mode(linker=mpi_linker)