From 28e9c79fcd7ed8798caaca600693cb3f3ffc29b8 Mon Sep 17 00:00:00 2001
From: Alex Lee <alexleegk@gmail.com>
Date: Wed, 11 Apr 2018 21:22:36 -0700
Subject: [PATCH] Add imports from future in main scripts.

---
 scripts/combine_results.py | 4 ++++
 scripts/evaluate.py        | 4 ++++
 scripts/generate.py        | 5 ++++-
 scripts/plot_results.py    | 4 ++++
 scripts/train.py           | 4 ++++
 5 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/scripts/combine_results.py b/scripts/combine_results.py
index b6251590..2b684ac1 100644
--- a/scripts/combine_results.py
+++ b/scripts/combine_results.py
@@ -1,3 +1,7 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
 import argparse
 import glob
 import itertools
diff --git a/scripts/evaluate.py b/scripts/evaluate.py
index b2932022..a19d6c9f 100644
--- a/scripts/evaluate.py
+++ b/scripts/evaluate.py
@@ -1,3 +1,7 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
 import argparse
 import csv
 import errno
diff --git a/scripts/generate.py b/scripts/generate.py
index 89e66fbf..9d080a77 100644
--- a/scripts/generate.py
+++ b/scripts/generate.py
@@ -1,3 +1,7 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
 import argparse
 import errno
 import json
@@ -10,7 +14,6 @@ import tensorflow as tf
 
 from video_prediction import datasets, models
 from video_prediction.utils.ffmpeg_gif import save_gif
-from tensorflow.python.util import nest
 
 
 def main():
diff --git a/scripts/plot_results.py b/scripts/plot_results.py
index db85f0f7..d0d25d14 100644
--- a/scripts/plot_results.py
+++ b/scripts/plot_results.py
@@ -1,3 +1,7 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
 import argparse
 import glob
 import os
diff --git a/scripts/train.py b/scripts/train.py
index 606a426d..c8035095 100644
--- a/scripts/train.py
+++ b/scripts/train.py
@@ -1,3 +1,7 @@
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
 import argparse
 import errno
 import itertools
-- 
GitLab