From f7bf106d77c152d9dad0503bdd43f2afa090261a Mon Sep 17 00:00:00 2001
From: Tom Le Feuvre <tjlefeuvre@gmail.com>
Date: Tue, 13 Apr 2021 12:36:49 +0200
Subject: [PATCH] Add pytest to Python for ML packages

---
 extra/Dockerfile                     | 1 +
 extra/tests/python3-for-ml/script.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/extra/Dockerfile b/extra/Dockerfile
index 689d3ee..07dea92 100644
--- a/extra/Dockerfile
+++ b/extra/Dockerfile
@@ -8,6 +8,7 @@ RUN apt-get update && \
         numpy \
         pandas \
         scikit-learn \
+        pytest \
         scipy && \
     rm -rf /var/lib/apt/lists/*
 
diff --git a/extra/tests/python3-for-ml/script.py b/extra/tests/python3-for-ml/script.py
index 01e6b70..f85847a 100644
--- a/extra/tests/python3-for-ml/script.py
+++ b/extra/tests/python3-for-ml/script.py
@@ -1,6 +1,7 @@
 import mlxtend
 import numpy
 import pandas
+import pytest
 import scipy
 import sklearn