main.yml

---
# roles/ml-python-packages

# This should be optimized so that only one check / install for pip3 is needed
# Still we should make sure that pip3 is actually installed before trying to use it
- name: Install dependecies
  become: true
  apt:
    name:
      - python3-pip
      - libatlas-base-dev
    state: present
    update_cache: true

- name: Remove conflicting apt packages
  become: true
  apt:
    name:
      - python3-pil
      - python3-numpy
    state: absent

- name: Install ml python packages
  become: true
  pip:
    executable: pip3
    name:
      - numpy
      - sklearn
      - matplotlib
      - pandas
      - Pillow
      # - torch
      # - torchvision
      # - fastai
    state: present