From b8bf706976720527b455eb665fe94f907e372b65 Mon Sep 17 00:00:00 2001
From: Patrick Simianer <p@simianer.de>
Date: Wed, 13 Nov 2013 18:00:10 +0100
Subject: unit tests for extractor loo sampling

---
 extractor/mocks/mock_rule_factory.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'extractor/mocks/mock_rule_factory.h')

diff --git a/extractor/mocks/mock_rule_factory.h b/extractor/mocks/mock_rule_factory.h
index 7389b396..86a084b5 100644
--- a/extractor/mocks/mock_rule_factory.h
+++ b/extractor/mocks/mock_rule_factory.h
@@ -7,7 +7,7 @@ namespace extractor {
 
 class MockHieroCachingRuleFactory : public HieroCachingRuleFactory {
  public:
-  MOCK_METHOD1(GetGrammar, Grammar(const vector<int>& word_ids));
+  MOCK_METHOD3(GetGrammar, Grammar(const vector<int>& word_ids, const unordered_set<int> blacklisted_sentence_ids, const shared_ptr<DataArray> source_data_array));
 };
 
 } // namespace extractor
-- 
cgit v1.2.3