Added missing import

tensorflow · Feb 4, 2019 · 1ca9496 · 1ca9496
1 parent e4ae074
commit 1ca9496
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/optvis/test_objectives.py b/tests/optvis/test_objectives.py
@@ -5,6 +5,8 @@
 import tensorflow as tf
 import numpy as np
 from lucid.optvis import objectives, param, render, transform
+from lucid.optvis.objectives import wrap_objective
+
 
 np.random.seed(42)