refactor: make OutputFeature.predictions() name consistent to origina…

…l naming
ludwig-ai · Mar 25, 2020 · 9e36022 · 9e36022
1 parent 392fbf4
commit 9e36022
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ludwig/features/numerical_feature.py b/ludwig/features/numerical_feature.py
@@ -172,7 +172,7 @@ def logits(
 
         return predictions, inputs
 
-    def predict(
+    def predictions(
             self,
             inputs,  # hidden
     ):

diff --git a/ludwig/models/ecd.py b/ludwig/models/ecd.py
@@ -85,7 +85,7 @@ def predict(self, inputs, output_features=None):
             )
 
         for of_name in of_list:
-            predictions = self.output_features[of_name].predict(logits[of_name])
+            predictions = self.output_features[of_name].predictions(logits[of_name])
 
         return predictions