Fix scale_block for constant blocks

rgcca-factory · Dec 14, 2024 · 1cc332e · 1cc332e
1 parent 0b1f2eb
commit 1cc332e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/R/scale_inertia.R b/R/scale_inertia.R
@@ -9,6 +9,9 @@ scale_inertia <- function(A, sqrt_N, scale, na.rm) {
     z <- A
   }
   fac <- 1 / sqrt_N * norm(z, type = "F")
+  if (fac == 0) {
+    return(A)
+  }
   y <- A / fac
   if (scale) {
     attr(y, "scaled:scale") <- attr(A, "scaled:scale") * fac

diff --git a/R/scale_lambda1.R b/R/scale_lambda1.R
@@ -15,6 +15,9 @@ scale_lambda1 <- function(A, sqrt_N, scale, na.rm) {
       symmetric = TRUE, only.values = TRUE
     )$values[1]
   ))
+  if (lambda == 0) {
+    return(A)
+  }
   y <- A / lambda
   if (scale) {
     attr(y, "scaled:scale") <- attr(A, "scaled:scale") * lambda