04238fb26bf4f0555928a7e00d78cff5a1b7cd6e,tensorflow_ranking/python/keras/network_test.py,UnivariateRankingNetworkTest,setUp,#UnivariateRankingNetworkTest#,123

Before Change

  def setUp(self):
    super(UnivariateRankingNetworkTest, self).setUp()
    self.context_feature_columns = _context_feature_columns()
    self.example_feature_columns = _example_feature_columns()
    self.features = _features()
    self.ranker = _DummyUnivariateRankingNetwork(
        context_feature_columns=self.context_feature_columns,
        example_feature_columns=self.example_feature_columns)

After Change

  def setUp(self):
    super(UnivariateRankingNetworkTest, 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
    self._features = _features()
    self._ranker = _DummyUnivariateRankingNetwork(
        context_feature_columns=self._context_feature_columns,
        example_feature_columns=self._example_feature_columns)

Instances