Add py_to_r conversion methods for TF 2.9 tracking wrappers

rstudio · Feb 1, 2024 · 15ee3a1 · 15ee3a1
1 parent 158bf9c
commit 15ee3a1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -105,6 +105,8 @@ S3method(prod,tensorflow.tensor)
 S3method(py_str,tensorflow.python.framework.tensor_shape.TensorShape)
 S3method(py_to_r,tensorflow.python.trackable.data_structures.ListWrapper)
 S3method(py_to_r,tensorflow.python.trackable.data_structures._DictWrapper)
+S3method(py_to_r,tensorflow.python.training.tracking.data_structures.ListWrapper)
+S3method(py_to_r,tensorflow.python.training.tracking.data_structures._DictWrapper)
 S3method(range,tensorflow.tensor)
 S3method(rbind,tensorflow.tensor)
 S3method(rep,tensorflow.tensor)

diff --git a/R/generics.R b/R/generics.R
@@ -855,3 +855,11 @@ py_to_r.tensorflow.python.trackable.data_structures.ListWrapper <- function(x) {
   import_builtins()$list(x)
 }
 
+# tf_version <= 2.9
+#' @export
+py_to_r.tensorflow.python.training.tracking.data_structures.ListWrapper <-
+  py_to_r.tensorflow.python.trackable.data_structures.ListWrapper
+
+#' @export
+py_to_r.tensorflow.python.training.tracking.data_structures._DictWrapper <-
+  py_to_r.tensorflow.python.trackable.data_structures._DictWrapper