Update custom.py

david-thrower · web-flow · commit 5e583c76dddd · 2024-06-13T14:56:00.000-04:00
Fix missing import.
diff --git a/cerebros/custom/custom.py b/cerebros/custom/custom.py
@@ -1,4 +1,6 @@
 
+import tensorflow as tf
+
 # Custom keras layer, analogue of Dense that does 
 # ternary mathematical operations 
 class TernaryDenseLayer(tf.keras.layers.Layer):
