Merge branch 'main' into dependabot-pip-wandb-eq-0.15.star

manujosephv · Jul 12, 2023 · 6f3c84a · 6f3c84a
2 parents 0708b90 + 8af2e38
commit 6f3c84a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -192,6 +192,13 @@ loaded_model = TabularModel.load_from_checkpoint("examples/basic")
         </a>
     </td></tr>
 <tr>
+    <td align="center">
+        <a href="https://github.com/EeyoreLee">
+            <img src="https://avatars.githubusercontent.com/u/49790022?v=4" width="100;" alt="EeyoreLee"/>
+            <br />
+            <sub><b>Earlee</b></sub>
+        </a>
+    </td>
     <td align="center">
         <a href="https://github.com/JulianRein">
             <img src="https://avatars.githubusercontent.com/u/35046938?v=4" width="100;" alt="JulianRein"/>

diff --git a/src/pytorch_tabular/models/base_model.py b/src/pytorch_tabular/models/base_model.py
@@ -103,7 +103,7 @@ def __init__(
                     config.metrics.append(metric.__name__)
                     config.metrics_params.append(vars(metric))
             if config.task == "classification":
-                config.metrics_prob_inputs = self.custom_metrics_prob_inputs
+                config.metrics_prob_input = self.custom_metrics_prob_inputs
         # Updating default metrics in config
         elif config.task == "classification":
             # Adding metric_params to config for classification task