From 86f9eb76cc3e99072f1314a2a82d3406d70f2e1e Mon Sep 17 00:00:00 2001
From: Rui Vieira <ruidevieira@googlemail.com>
Date: Mon, 9 Sep 2024 22:49:21 +0100
Subject: [PATCH] Update TrustyAI

- Update TrustyAI to 0.6.1 (adds support for TMaRCo local models)
- Add PyTorch, transformers, datasets and accelerate as TrustyAI notebook image dependencies
---
 jupyter/trustyai/ubi9-python-3.11/Pipfile | 12 +++++++++++-
 jupyter/trustyai/ubi9-python-3.9/Pipfile  | 12 +++++++++++-
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/jupyter/trustyai/ubi9-python-3.11/Pipfile b/jupyter/trustyai/ubi9-python-3.11/Pipfile
index 31fd97a9e..05c6e2771 100644
--- a/jupyter/trustyai/ubi9-python-3.11/Pipfile
+++ b/jupyter/trustyai/ubi9-python-3.11/Pipfile
@@ -3,11 +3,21 @@ url = "https://pypi.org/simple"
 verify_ssl = true
 name = "pypi"
 
+[[source]]
+url = "https://download.pytorch.org/whl/cu121"
+verify_ssl = true
+name = "pytorch"
+
 [dev-packages]
 
 [packages]
+# PyTorch packages
+torch = {version = "~=2.2.1", index = "pytorch"}
 # TrustyAI packages
-trustyai = "~=0.6.0"
+transformers = "~=4.36.2"
+datasets = "~=2.21.0"
+accelerate = "~=0.34.2"
+trustyai = "~=0.6.1"
 # Datascience and useful extensions
 boto3 = "~=1.34.50"
 kafka-python = "~=2.0.2"
diff --git a/jupyter/trustyai/ubi9-python-3.9/Pipfile b/jupyter/trustyai/ubi9-python-3.9/Pipfile
index 35cccc53b..b45c5768a 100644
--- a/jupyter/trustyai/ubi9-python-3.9/Pipfile
+++ b/jupyter/trustyai/ubi9-python-3.9/Pipfile
@@ -3,11 +3,21 @@ url = "https://pypi.org/simple"
 verify_ssl = true
 name = "pypi"
 
+[[source]]
+url = "https://download.pytorch.org/whl/cu121"
+verify_ssl = true
+name = "pytorch"
+
 [dev-packages]
 
 [packages]
+# PyTorch packages
+torch = {version = "~=2.2.1", index = "pytorch"}
 # TrustyAI packages
-trustyai = "~=0.6.0"
+transformers = "~=4.36.2"
+datasets = "~=2.21.0"
+accelerate = "~=0.34.2"
+trustyai = "~=0.6.1"
 # Datascience and useful extensions
 boto3 = "~=1.34.50"
 kafka-python = "~=2.0.2"