Bug Fix in Classifier test

FBerding · Oct 1, 2023 · bafa650 · bafa650
1 parent ac96018
commit bafa650
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git 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
@@ -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,