From a007a090e3f105ef5a983f886aa2ad5c76d02302 Mon Sep 17 00:00:00 2001
From: Vincent Roseberry <rosbo@google.com>
Date: Tue, 26 Jan 2021 18:27:15 +0000
Subject: [PATCH] Add flask.

- Also added tests to prevent regression.

flask was removed from the base image we use:

```
rosbo@rosbo:~$ docker run -it --rm gcr.io/deeplearning-platform-release/base-cpu:m46 pip show flask
Name: Flask
Version: 1.1.2
Summary: A simple framework for building complex web applications.
Home-page: https://palletsprojects.com/p/flask/
Author: Armin Ronacher
Author-email: armin.ronacher@active-4.com
License: BSD-3-Clause
Location: /opt/conda/lib/python3.7/site-packages
Requires: Werkzeug, itsdangerous, click, Jinja2
Required-by:
rosbo@rosbo:~$ docker run -it --rm gcr.io/deeplearning-platform-release/base-cpu:m61 pip show flask
WARNING: Package(s) not found: flask
```
---
 Dockerfile          |  1 +
 tests/test_flask.py | 10 ++++++++++
 2 files changed, 11 insertions(+)
 create mode 100644 tests/test_flask.py

diff --git a/Dockerfile b/Dockerfile
index c25bd1e0..61bb70b3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -413,6 +413,7 @@ RUN pip install flashtext && \
     pip install pytorch-lightning && \
     pip install datatable && \
     pip install sympy && \
+    pip install flask && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages
diff --git a/tests/test_flask.py b/tests/test_flask.py
new file mode 100644
index 00000000..7b0ed27c
--- /dev/null
+++ b/tests/test_flask.py
@@ -0,0 +1,10 @@
+import unittest
+
+from flask import Flask, request
+
+class TestFlask(unittest.TestCase):
+    def test_request(self):
+        app = Flask(__name__)
+        with app.test_request_context('/foo', method='POST'):
+            assert request.path == '/foo'
+            assert request.method == 'POST'