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 @@ ...@@ -5,7 +5,7 @@
import tensorflow as tf import tensorflow as tf
try: try:
from tensorflow.contrib.staging import StagingArea from tensorflow.python.ops.data_flow_ops import StagingArea
except ImportError: except ImportError:
pass 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