Skip to content

Commit

Permalink
add S3 py_to_r methods for Tracked{Dict,List,Set}
Browse files Browse the repository at this point in the history
  • Loading branch information
t-kalinowski committed Feb 1, 2024
1 parent 4b9f87e commit a9ecac4
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 0 deletions.
3 changes: 3 additions & 0 deletions NAMESPACE
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,9 @@ S3method(print,keras.src.models.model.Model)
S3method(print,keras_shape)
S3method(print,keras_training_history)
S3method(py_str,keras.src.models.model.Model)
S3method(py_to_r,keras.src.utils.tracking.TrackedDict)
S3method(py_to_r,keras.src.utils.tracking.TrackedList)
S3method(py_to_r,keras.src.utils.tracking.TrackedSet)
S3method(py_to_r,tensorflow.python.ops.gen_linalg_ops.Qr)
S3method(py_to_r,tensorflow.python.ops.gen_nn_ops.TopKV2)
S3method(py_to_r_wrapper,keras.src.initializers.initializer.Initializer)
Expand Down
12 changes: 12 additions & 0 deletions R/s3-methods.R
Original file line number Diff line number Diff line change
Expand Up @@ -30,3 +30,15 @@ as.double.keras.src.backend.common.variables.KerasVariable <- function(x, ...) {
as.integer.keras.src.backend.common.variables.KerasVariable <- function(x, ...) {
as.integer(as_r_value(keras$ops$convert_to_numpy(x)))
}

## May need to revisit this; either to disable it, or export a custom $<- method
## for base classes like Layer, so that compound assignment expressions aren't a
## problem.
#' @export
py_to_r.keras.src.utils.tracking.TrackedDict <- function(x) import("builtins")$dict(x)

#' @export
py_to_r.keras.src.utils.tracking.TrackedList <- function(x) import("builtins")$list(x)

#' @export
py_to_r.keras.src.utils.tracking.TrackedSet <- function(x) import("builtins")$list(x)

0 comments on commit a9ecac4

Please sign in to comment.