code: def make_estimator_input_fn(self,
                                mode,
                                hparams,
                                data_dir=None,
                                force_repeat=False,
                                prevent_repeat=False,
                                dataset_kwargs=None):
      
  
      def estimator_input_fn(params, config):
        return self.input_fn(
            mode,
            hparams,
            data_dir=data_dir,
            params=params,
            config=config,
            force_repeat=force_repeat,
            prevent_repeat=prevent_repeat,
            dataset_kwargs=dataset_kwargs)
  
      return estimator_input_fn
docs: Return input_fn wrapped for Estimator.