Fixed relative imports

ryanjulian · Jun 7, 2018 · c7bc9d6 · c7bc9d6
1 parent edfc757
commit c7bc9d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/sandbox/rocky/tf/algos/batch_polopt.py b/sandbox/rocky/tf/algos/batch_polopt.py
@@ -124,8 +124,7 @@ def train(self, sess=None):
                 logger.log("Optimizing policy...")
                 self.optimize_policy(itr, samples_data)
                 logger.log("Saving snapshot...")
-                params = self.get_itr_snapshot(itr,
-                                               samples_data)  # , **kwargs)
+                params = self.get_itr_snapshot(itr, samples_data)
                 if self.store_paths:
                     params["paths"] = samples_data["paths"]
                 logger.save_itr_params(itr, params)

diff --git a/sandbox/rocky/tf/plotter/__init__.py b/sandbox/rocky/tf/plotter/__init__.py
@@ -1 +1 @@
-from sandbox.rocky.tf.plotter.plotter import *
+from sandbox.rocky.tf.plotter.plotter import Plotter