more formatting cleanup (#1015)

* more formatting cleanup Signed-off-by: Niels Bantilan <[email protected]> * more cleanup Signed-off-by: Niels Bantilan <[email protected]> * use requirements.in for mlflow plugin example Signed-off-by: Niels Bantilan <[email protected]> --------- Signed-off-by: Niels Bantilan <[email protected]> Co-authored-by: Kevin Su <[email protected]>
flyteorg · Jul 6, 2023 · 80427a5 · 80427a5
1 parent 73840e6
commit 80427a5
Show file tree

Hide file tree

Showing 5 changed files with 419 additions and 4 deletions.
diff --git a/examples/athena_plugin/athena_plugin/athena.py b/examples/athena_plugin/athena_plugin/athena.py
@@ -86,11 +86,11 @@ def no_io_wf():
     SELECT * FROM vaccinations where iso_code like  '{{ .Inputs.iso_code }}'
     """,
     # While we define a generic schema as the output here, Flyte supports more strongly typed schemas to provide
-    # better compile-time checks for task compatibility. Refer to :py:class:`flytekit.FlyteSchema` for more details
+    # better compile-time checks for task compatibility. Refer to {py:class}`flytekit.FlyteSchema` for more details
     output_schema_type=FlyteSchema,
     # Cache the output data so we don't have to re-run the query in future workflow iterations
     # should we decide to change how we manipulate data downstream.
-    # For more information about caching, visit :ref:`Task Caching <task_cache>`
+    # For more information about caching, visit {ref}`Task Caching <task_cache>`
     cache=True,
     cache_version="1.0",
 )

diff --git a/examples/mlflow_plugin/Dockerfile b/examples/mlflow_plugin/Dockerfile
@@ -24,7 +24,8 @@ RUN python3 -m venv ${VENV}
 ENV PATH="${VENV}/bin:$PATH"
 
 # Install Python dependencies
-RUN pip install flytekitplugins-mlflow tensorflow protobuf==3.20.1
+COPY requirements.txt /root
+RUN pip install -r /root/requirements.txt
 
 # Copy the actual code
 COPY . /root/

diff --git a/examples/mlflow_plugin/requirements.in b/examples/mlflow_plugin/requirements.in
@@ -0,0 +1,3 @@
+flytekitplugins-mlflow
+tensorflow
+protobuf==3.20.1