From bafa6504f45aff2ab124641a78fc56746e4ee472 Mon Sep 17 00:00:00 2001
From: Florian Berding <florian.berding@uni-hamburg.de>
Date: Sun, 1 Oct 2023 20:20:19 +0200
Subject: [PATCH] Bug Fix in Classifier test

---
 R/install_and_config.R                         | 6 ++++--
 tests/testthat/test-14_classifier_neural_net.R | 3 ---
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/R/install_and_config.R b/R/install_and_config.R
index 20f28d9..b21a489 100644
--- a/R/install_and_config.R
+++ b/R/install_and_config.R
@@ -22,7 +22,8 @@ install_py_modules<-function(envname="aifeducation",
                       "tokenizers",
                       "datasets",
                       "codecarbon",
-                      "accelerate")
+                      "accelerate"
+                      )
 
   if(remove_first==TRUE){
     conda_environments<-reticulate::conda_list()
@@ -40,7 +41,8 @@ install_py_modules<-function(envname="aifeducation",
     packages = c(
       paste0("tensorflow",tf_version),
       "torch",
-      "keras"),
+      "keras",
+      "cudatoolkit"),
     envname = envname,
     conda = "auto",
     pip = TRUE
diff --git a/tests/testthat/test-14_classifier_neural_net.R b/tests/testthat/test-14_classifier_neural_net.R
index 2ef9562..74d8be7 100644
--- a/tests/testthat/test-14_classifier_neural_net.R
+++ b/tests/testthat/test-14_classifier_neural_net.R
@@ -174,13 +174,10 @@ for (n_classes in 2:3){
       rec_act_fct="tanh")
     expect_s3_class(classifier,
                     class="TextEmbeddingClassifierNeuralNet")
-  })
 
-  test_that("te_classifier_sustainability_start",{
     expect_false(classifier$get_sustainability_data()$sustainability_tracked)
   })
 
-
   #-------------------------------------------------------------------------------
   classifier<-TextEmbeddingClassifierNeuralNet$new(
     ml_framework = ml_framework,