Commit acec5ac2 authored by Aaron Gokaslan's avatar Aaron Gokaslan Committed by Yuxin Wu

Corrected tensorflow path to stagingarea (#275)

Bugfix for StagingArea. It appears as though Tensorflow is not adding it in the old contrib location. Should be backwards compatible.
parent bbe7a12b
......@@ -5,7 +5,7 @@
import tensorflow as tf
try:
from tensorflow.contrib.staging import StagingArea
from tensorflow.python.ops.data_flow_ops import StagingArea
except ImportError:
pass
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment