fix k_set_epsilon() and k_set_floatx() for tf

Author: kevinykuo

R/backend.R

@@ -867,9 +867,7 @@ k_epsilon <- function() {
 #' @rdname k_epsilon
 #' @export
 k_set_epsilon <- function(e) {
-  keras$backend$set_epsilon(
-    e = e
-  )
+  keras$backend$set_epsilon(e)
 }
 
 
@@ -994,9 +992,7 @@ k_floatx <- function() {
 #' @rdname k_floatx
 #' @export
 k_set_floatx <- function(floatx) {
-  keras$backend$set_floatx(
-    floatx = floatx
-  )
+  keras$backend$set_floatx(floatx)
 }
 
 

tests/testthat/test-backend.R

@@ -115,3 +115,16 @@ test_backend("k_concatenate", {
   k_concatenate(list(x, y), 1)
 })
 
+test_backend("k_set_floatx", {
+  skip_if_cntk()
+  skip_if_theano()
+  k_set_floatx("float16")
+  expect_identical(k_floatx(), "float16")
+})
+
+test_backend("k_set_epsilon", {
+  skip_if_cntk()
+  skip_if_theano()
+  k_set_epsilon(1e-4)
+  expect_equal(k_epsilon(), 1e-4)
+})