install module yml

.gitlab-ci.yml

@@ -9,12 +9,14 @@ stages:
 build:
   stage: build
   script:
+    - pip install -r requirements.txt
     - python machine_learning.py
 
 # Test stage
 test:
   stage: test
   script:
+    - pip install -r requirements.txt
     - pytest test_prediction.py
 
 # Export stage

requirements.txt

+numpy
+matplotlib
+scikit-learn
+pytest
+joblib
\ No newline at end of file