From 4b5171682c00aa020e0fb8237915d616d1f8e105 Mon Sep 17 00:00:00 2001
From: Felipe <lipecaso@gmail.com>
Date: Wed, 25 Oct 2023 20:17:27 +0000
Subject: [PATCH] add gpu_memory parameter in sits_classify.sits functionn

---
 R/sits_classify.R    | 2 ++
 man/sits_classify.Rd | 5 +++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/R/sits_classify.R b/R/sits_classify.R
index 57fb11272..c3d43699f 100644
--- a/R/sits_classify.R
+++ b/R/sits_classify.R
@@ -165,6 +165,7 @@ sits_classify.sits <- function(data,
                                ...,
                                filter_fn = NULL,
                                multicores = 2L,
+                               gpu_memory = NULL,
                                progress = TRUE) {
     # Pre-conditions
     data <- .check_samples_ts(data)
@@ -181,6 +182,7 @@ sits_classify.sits <- function(data,
         ml_model = ml_model,
         filter_fn = filter_fn,
         multicores = multicores,
+        gpu_memory = gpu_memory,
         progress = progress
     )
     return(classified_ts)
diff --git a/man/sits_classify.Rd b/man/sits_classify.Rd
index 87ea90f5f..08307dcb4 100644
--- a/man/sits_classify.Rd
+++ b/man/sits_classify.Rd
@@ -25,6 +25,7 @@ sits_classify(
   ...,
   filter_fn = NULL,
   multicores = 2L,
+  gpu_memory = NULL,
   progress = TRUE
 )
 
@@ -84,6 +85,8 @@ sits_classify(
 
 \item{progress}{Logical: Show progress bar?}
 
+\item{gpu_memory}{Memory available in GPU (default = NULL)}
+
 \item{roi}{Region of interest (either an sf object, shapefile,
 or a numeric vector with named XY values
 ("xmin", "xmax", "ymin", "ymax") or
@@ -99,8 +102,6 @@ named lat/long values
 \item{memsize}{Memory available for classification in GB
 (integer, min = 1, max = 16384).}
 
-\item{gpu_memory}{Memory available in GPU (default = NULL)}
-
 \item{output_dir}{Valid directory for output file.
 (character vector of length 1).}