[CI] Fix website build (#1573)

dmlc · May 21, 2021 · e78f310 · e78f310
1 parent df52171
commit e78f310
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -250,7 +250,7 @@ def setup(app):
     'examples_dirs': [],
     'gallery_dirs': [],
     'subsection_order': ExplicitOrder([]),
-    'find_mayavi_figures': False,
+    'image_scrapers': ('matplotlib', 'mayavi'),
     'filename_pattern': '.py',
     'expected_failing_examples': []
 }

diff --git a/src/gluonnlp/attention_cell.py b/src/gluonnlp/attention_cell.py
@@ -15,6 +15,12 @@
 # specific language governing permissions and limitations
 # under the License.
 """Attention cells."""
+
+__all__ = ['gen_self_attn_mask', 'gen_mem_attn_mask',
+           'masked_softmax', 'masked_logsoftmax', 'multi_head_dot_attn',
+           'MultiHeadAttentionCell', 'gen_rel_position',
+           'RelAttentionScoreCell']
+
 import math
 import mxnet as mx
 from mxnet import np, npx