Skip to content

Commit 8ca9f7a

Browse files
committed
use default_version from keras in install_keras()
Previously, would default to use the tensorflow:::default_version
1 parent 883996a commit 8ca9f7a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

R/install.R

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,9 @@ install_keras <- function(method = c("auto", "virtualenv", "conda"),
3030
if(!is.null(extra_packages))
3131
pkgs[gsub("[=<>~]{1,2}[0-9.]+$", "", extra_packages)] <- extra_packages
3232

33+
if(tensorflow == "default") # may be different from tensorflow
34+
tensorflow <- default_version
35+
3336
tensorflow::install_tensorflow(
3437
method = match.arg(method),
3538
conda = conda,
@@ -39,7 +42,7 @@ install_keras <- function(method = c("auto", "virtualenv", "conda"),
3942
)
4043
}
4144

42-
default_version = numeric_version("2.6")
45+
default_version <- numeric_version("2.6")
4346

4447
default_extra_packages <- function(tensorflow_version) {
4548
pkgs <- c("tensorflow-hub", "scipy", "requests", "pyyaml", "Pillow", "h5py", "pandas")

0 commit comments

Comments
 (0)