change to use underlying class

Sigrid Keydana · Sigrid Keydana · commit 87b85d57b230 · 2019-02-01T20:38:12.000+01:00
diff --git a/R/model.R b/R/model.R
@@ -973,7 +973,8 @@ is_main_thread_generator.keras_preprocessing.image.Iterator <- function(x) {
 }
 
 is_tensorflow_dataset <- function(x) {
-  inherits(x, "tf_dataset")
+  inherits(x, "tensorflow.python.data.ops.dataset_ops.DatasetV2") ||
+    inherits(x, "tensorflow.python.data.ops.dataset_ops.Dataset")
 }
 
 resolve_tensorflow_dataset <- function(x) {

Original file line number	Diff line number	Diff line change
`@@ -973,7 +973,8 @@ is_main_thread_generator.keras_preprocessing.image.Iterator <- function(x) {`
`973`	`973`	`}`
`974`	`974`
`975`	`975`	`is_tensorflow_dataset <- function(x) {`
`976`		`- inherits(x, "tf_dataset")`
	`976`	`+ inherits(x, "tensorflow.python.data.ops.dataset_ops.DatasetV2") \|\|`
	`977`	`+ inherits(x, "tensorflow.python.data.ops.dataset_ops.Dataset")`
`977`	`978`	`}`
`978`	`979`
`979`	`980`	`resolve_tensorflow_dataset <- function(x) {`