diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 44100f7e..d902d404 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -498,6 +498,7 @@ RUN rm -rf /opt/conda/lib/python3.10/site-packages/numpy* && \
         # pycrypto is used by competitions team.
         pycryptodome \
 	nbdev \
+	umap-learn \
         easyocr \
         onnx \
         tables \
diff --git a/tests/test_umap_learn.py b/tests/test_umap_learn.py
new file mode 100644
index 00000000..a49b40e6
--- /dev/null
+++ b/tests/test_umap_learn.py
@@ -0,0 +1,7 @@
+import unittest
+
+import umap
+
+class TestUmapLearn(unittest.TestCase):
+    def test(self):
+        self.assertGreater(len(umap.__version__), 0)