fix: remove Model as suffix in XGB model name, update version to 2.1.1 (

#191)
h1st-ai · May 5, 2023 · 892a8dc · 892a8dc
1 parent 214f6d1
commit 892a8dc
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/h1st/model/ml/xgboost/classifier.py b/h1st/model/ml/xgboost/classifier.py
@@ -2,7 +2,7 @@
 
 
 class XGBClassifierModel(XGBRegressionModel):
-    name = 'XGBClassifierModel'
+    name = 'XGBClassifier'
 
     def predict(self, input_data: dict) -> dict:
         pred = super().predict(input_data)[self.output_key]

diff --git a/h1st/model/ml/xgboost/regression.py b/h1st/model/ml/xgboost/regression.py
@@ -16,7 +16,7 @@ class XGBRegressionModel(Model):
 
     input_key = 'X'
     output_key = 'predictions'
-    name = 'XGBRegressionModel'
+    name = 'XGBRegression'
 
     def __init__(
         self,

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "h1st"
-version = "2.1.0"
+version = "2.1.1"
 description = "Human-First AI (H1st)"
 authors = ["Aitomatic, Inc. <engineering@aitomatic.com>"]
 license = "Apache-2.0"

diff --git a/setup.cfg b/setup.cfg
@@ -4,4 +4,4 @@
 
 [metadata]
 name = h1st
-version = 2.1.0
+version = 2.1.1