Index: components/translate/core/browser/translate_ranker_impl.cc |
diff --git a/components/translate/core/browser/translate_ranker_impl.cc b/components/translate/core/browser/translate_ranker_impl.cc |
index b1d587f6c88133a015b11757ea18dba8fb76d4cd..64fab4535af5b9f924b457599c837162ec50a24a 100644 |
--- a/components/translate/core/browser/translate_ranker_impl.cc |
+++ b/components/translate/core/browser/translate_ranker_impl.cc |
@@ -18,10 +18,11 @@ |
#include "base/strings/string_util.h" |
#include "base/task_runner.h" |
#include "base/threading/thread_task_runner_handle.h" |
+#include "components/machine_intelligence/proto/ranker_model.pb.h" |
+#include "components/machine_intelligence/proto/translate_ranker_model.pb.h" |
+#include "components/machine_intelligence/ranker_model.h" |
#include "components/metrics/proto/translate_event.pb.h" |
-#include "components/translate/core/browser/proto/ranker_model.pb.h" |
-#include "components/translate/core/browser/proto/translate_ranker_model.pb.h" |
-#include "components/translate/core/browser/ranker_model.h" |
+#include "components/translate/core/browser/translate_download_manager.h" |
#include "components/translate/core/common/translate_switches.h" |
#include "components/ukm/public/ukm_entry_builder.h" |
#include "components/ukm/public/ukm_recorder.h" |
@@ -32,9 +33,10 @@ namespace translate { |
namespace { |
-using chrome_intelligence::RankerModel; |
-using chrome_intelligence::RankerModelProto; |
-using chrome_intelligence::TranslateRankerModel; |
+using machine_intelligence::RankerModel; |
+using machine_intelligence::RankerModelProto; |
+using machine_intelligence::TranslateRankerModel; |
+using machine_intelligence::RankerModelStatus; |
const double kTranslationOfferDefaultThreshold = 0.5; |
@@ -141,11 +143,12 @@ TranslateRankerImpl::TranslateRankerImpl(const base::FilePath& model_path, |
translate::kTranslateRankerDecisionOverride)), |
weak_ptr_factory_(this) { |
if (is_query_enabled_ || is_enforcement_enabled_) { |
- model_loader_ = base::MakeUnique<RankerModelLoader>( |
+ model_loader_ = base::MakeUnique<machine_intelligence::RankerModelLoader>( |
base::Bind(&ValidateModel), |
base::Bind(&TranslateRankerImpl::OnModelAvailable, |
weak_ptr_factory_.GetWeakPtr()), |
- model_path, model_url, kUmaPrefix); |
+ TranslateDownloadManager::GetInstance()->request_context(), model_path, |
+ model_url, kUmaPrefix); |
// Kick off the initial load from cache. |
model_loader_->NotifyOfRankerActivity(); |
} |