RuntimeError: Evaluation error: object 'feature_names' not found.

[ds-kalle]

Hello,

I am trying to get the the tfestimators examples in https://tensorflow.rstudio.com/tools/tfdatasets/articles/introduction.html to work but always get an

 Error in py_call_impl(callable, dots$args, dots$keywords) : 
  RuntimeError: Evaluation error: object 'feature_names' not found. 

The code I am using is the following (had to change column_numeric to tf$feature_column$numeric_column due to rstudio/tfestimators#173):

library(tfestimators)
library(tfdatasets)

mtcars_spec <- csv_record_spec("mtcars.csv")
dataset <- text_line_dataset("mtcars.csv", record_spec = mtcars_spec) %>% 
  dataset_batch(128) %>% 
  dataset_repeat(3)

cols <- feature_columns(
  tf$feature_column$numeric_column("mpg"),
  tf$feature_column$numeric_column("disp")
)

model <- linear_regressor(feature_columns = cols)

model %>% train(
  input_fn(dataset, features = c(mpg, disp), response = cyl)
)

I am getting this error with both 3.6 (TF v1.14.1-dev20190611 Nightly build because of tensorflow/tensorflow#23410) as well as 2.7 (TF v1.13.1) on Mac OSX 10.14.5.
Andy ideas?

Thanks a lot,
Kalle