diff --git a/‎Dockerfile‎
Lines changed: 1 addition & 0 deletions b/‎Dockerfile‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎tests/test_pytorch_ignite.py‎
Lines changed: 13 additions & 0 deletions b/‎tests/test_pytorch_ignite.py‎
Lines changed: 13 additions & 0 deletions
@@ -481,6 +481,7 @@ RUN pip install flashtext && \
     pip install plotly_express && \
     pip install albumentations && \
     pip install allennlp && \
+    pip install pytorch-ignite && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages
 
@@ -0,0 +1,13 @@
+import unittest
+
+from ignite.engine import Engine
+
+
+class TestPytorchIgnite(unittest.TestCase):
+    def test_engine(self):
+
+        def update_fn(engine, batch):
+            pass
+
+        engine = Engine(update_fn)
+        engine.run([0, 1, 2])