From 324bd07b28888e922b980d59ab835b1e9173bb2d Mon Sep 17 00:00:00 2001
From: Vincent Roseberry <rosbo@google.com>
Date: Thu, 14 Jan 2021 17:21:12 +0000
Subject: [PATCH] Add sympy package

Required by our Learn vision class.

http://b/177552728
---
 Dockerfile          | 1 +
 tests/test_sympy.py | 8 ++++++++
 2 files changed, 9 insertions(+)
 create mode 100644 tests/test_sympy.py

diff --git a/Dockerfile b/Dockerfile
index b200d97b..89bba22c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -410,6 +410,7 @@ RUN pip install flashtext && \
     pip install pydegensac && \
     pip install pytorch-lightning && \
     pip install datatable && \
+    pip install sympy && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages
diff --git a/tests/test_sympy.py b/tests/test_sympy.py
new file mode 100644
index 00000000..d25991dc
--- /dev/null
+++ b/tests/test_sympy.py
@@ -0,0 +1,8 @@
+import unittest
+
+import numpy as np
+import sympy
+
+class TestSympy(unittest.TestCase):
+    def test_matrix(self):
+        self.assertEqual((2, 2), sympy.Matrix([[0, 1], [1, 0]]).shape)
\ No newline at end of file