Add flask. (#946)

- 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: [email protected]
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
```

Author: rosbo

Dockerfile

@@ -413,6 +413,8 @@ RUN pip install flashtext && \
     pip install pytorch-lightning && \
     pip install datatable && \
     pip install sympy && \
+    # flask is used by agents in the simulation competitions.
+    pip install flask && \
     # pycrypto is used by competitions team.
     pip install pycrypto && \
     /tmp/clean-layer.sh

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'