From a3490936831abdc6a5515cc4cecd268843622d82 Mon Sep 17 00:00:00 2001 From: paramecinm Date: Mon, 19 Apr 2021 19:29:57 +0800 Subject: [PATCH] Fix potential typo in base_estimator::infer --- euler_estimator/python/base_estimator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/euler_estimator/python/base_estimator.py b/euler_estimator/python/base_estimator.py index de6f2c0..209d083 100644 --- a/euler_estimator/python/base_estimator.py +++ b/euler_estimator/python/base_estimator.py @@ -166,7 +166,7 @@ def infer(self): worker_idx = 0 out_idxs = [] out_embeddings = [] - for output in estimator.predict(input_fn=self.evaluate_input_fn): + for output in estimator.predict(input_fn=self.infer_input_fn): out_idxs.append(output['idx']) out_embeddings.append(output['embedding']) out_embeddings = np.asarray(out_embeddings)