04238fb26bf4f0555928a7e00d78cff5a1b7cd6e,tensorflow_ranking/python/keras/feature_test.py,EncodeListwiseFeaturesTest,setUp,#EncodeListwiseFeaturesTest#,96

Before Change

  def setUp(self):
    super(EncodeListwiseFeaturesTest, self).setUp()
    self.context_feature_columns = _context_feature_columns()
    self.example_feature_columns = _example_feature_columns()

    // Batch size = 2, list_size = 2.
    self.features = _features()
    self.listwise_dense_layer = feature.EncodeListwiseFeatures(

After Change

  def setUp(self):
    super(EncodeListwiseFeaturesTest, self).setUp()
    (context_feature_columns, example_feature_columns,
     custom_objects) = _get_feature_columns()
    self._context_feature_columns = context_feature_columns
    self._example_feature_columns = example_feature_columns
    self._custom_objects = custom_objects

Instances