add export

anas-rz · Feb 24, 2024 · e1e096f · e1e096f
1 parent 250fd1a
commit e1e096f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/k3_addons/layers/attention/parnet.py b/k3_addons/layers/attention/parnet.py
@@ -1,7 +1,9 @@
 from keras import layers, Sequential
 from k3_addons.layers.pooling.adaptive_pooling import AdaptiveAveragePool2D
+from k3_addons.api_export import k3_export
 
 
+@k3_export(path="k3_addons.layers.ParNetAttention")
 class ParNetAttention(layers.Layer):
     def __init__(self, activation="selu"):
         super().__init__()