updated setup_modules and add setup_gpu_modules

NAMESPACE

@@ -10,7 +10,9 @@ export(nlp_scores)
 export(plot_sim_emotions)
 export(punctuate)
 export(rag)
+export(setup_gpu_modules)
 export(setup_miniconda)
+export(setup_modules)
 export(simulate_video)
 export(transformer_scores)
 export(video_scores)

R/setup_gpu_modules.R

@@ -0,0 +1,40 @@
+#' Install GPU Python Modules
+#'
+#' @description Installs GPU modules for the {transforEmotion} conda environment
+#'
+#' @details Installs modules for miniconda using \code{\link[reticulate]{conda_install}}
+#'
+#' @author Alexander P. Christensen <alexpaulchristensen@gmail.com>
+#'
+#' @export
+#'
+# Install GPU modules
+# Updated 03.02.2024
+setup_gpu_modules <- function()
+{
+
+  # Set necessary modules
+  modules <- c(
+    "autoawq"
+  )
+
+  # Determine whether any modules need to be installed
+  installed_modules <- reticulate::py_list_packages(envname = "transforEmotion")
+
+  # Determine missing modules
+  missing_modules <- modules[!modules %in% installed_modules$package]
+
+  # Determine if modules need to be installed
+  if(length(missing_modules) != 0){
+
+    # Send message to user
+    message("\nInstalling modules for 'transforEmotion'...")
+
+    # Install modules
+    reticulate::conda_install(
+      "transforEmotion", packages = missing_modules, pip = TRUE
+    )
+
+  }
+
+}

R/setup_modules.R

@@ -6,19 +6,19 @@
 #'
 #' @author Alexander P. Christensen <alexpaulchristensen@gmail.com>
 #'
-#' @noRd
+#' @export
 #'
 # Install modules
-# Updated 28.01.2024
+# Updated 03.02.2024
 setup_modules <- function()
 {
 
   # Set necessary modules
   modules <- c(
-    "accelerate", "autoawq", "llama-index", "nltk",
+    "accelerate", "llama-index", "nltk",
     "opencv-python", "pandas", "pypdf",
     "pytube", "pytz", "qdrant-client",
-    "sentencepiece", "tensorflow", "torch", 
+    "sentencepiece", "tensorflow", "torch",
     "torchaudio", "torchvision", "transformers"
   )