update: model renamed to hierarchical_rf

alercebroker · Aug 26, 2021 · d676d52 · d676d52
1 parent 4cca0ee
commit d676d52
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lc_classifier/classifier/models.py b/lc_classifier/classifier/models.py
@@ -108,7 +108,7 @@ def load_model(self, directory: str) -> None:
 
 
 class HierarchicalRandomForest(BaseClassifier):
-    MODEL_NAME = "hierarchical_random_forest"
+    MODEL_NAME = "hierarchical_rf"
     MODEL_VERSION = "1.1.0"
     MODEL_VERSION_NAME = f"{MODEL_NAME}_{MODEL_VERSION}"
     MODEL_PICKLE_PATH = os.path.join(PICKLE_PATH, f"{MODEL_VERSION_NAME}")