From a6c239f5f6af73fb1a645565495d3525d2ad3f6f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20M=C3=BCller?= <daniel.q.mueller@stud.h-da.de>
Date: Wed, 24 Nov 2021 18:11:47 +0000
Subject: [PATCH] Add additional python deps & fix tf detection
- Add additional python dependencies
- Change tensorflow installed detection to not check for version 2.7 .
Since tf is installed from file, it doesn't register as version 2.7
---
roles/ml-python-packages/files/requirements.txt | 2 ++
roles/ml-python-packages/tasks/main.yml | 4 ++--
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/roles/ml-python-packages/files/requirements.txt b/roles/ml-python-packages/files/requirements.txt
index e0c328c..290a362 100644
--- a/roles/ml-python-packages/files/requirements.txt
+++ b/roles/ml-python-packages/files/requirements.txt
@@ -3,3 +3,5 @@ matplotlib==3.4.*
pandas==1.3.*
Pillow==8.4.*
scikit-learn==1.0.*
+seaborn==0.11.*
+plotly==5.4.*
diff --git a/roles/ml-python-packages/tasks/main.yml b/roles/ml-python-packages/tasks/main.yml
index 0c9078a..c28fe2a 100644
--- a/roles/ml-python-packages/tasks/main.yml
+++ b/roles/ml-python-packages/tasks/main.yml
@@ -23,8 +23,8 @@
-- name: Check if tensorflow is installed
- shell: ". /home/{{ target_user }}/.ml-venv/bin/activate && pip3 freeze | grep -q tensorflow==2.7"
+- name: Check if tensorflow is installed (from wheel file)
+ shell: ". /home/{{ target_user }}/.ml-venv/bin/activate && pip3 freeze | grep -q 'tensorflow @ file'"
register: tfinstalled
changed_when: "tfinstalled.rc != 0"
failed_when: false
--
GitLab