fix: 🐛 Fix depth_ property as an alias of states_

bayesclass/clfs.py

@@ -21,7 +21,6 @@ class BayesBase(BaseEstimator, ClassifierMixin):
         self.show_progress = show_progress
         # To keep compatiblity with the benchmark platform
         self.nodes_leaves = self.nodes_edges
-        self.depth_ = self.states_
 
     def _more_tags(self):
         return {
@@ -71,6 +70,10 @@ class BayesBase(BaseEstimator, ClassifierMixin):
             return sum([len(item) for _, item in self.model_.states.items()])
         return 0
 
+    @property
+    def depth_(self):
+        return self.states_
+
     def fit(self, X, y, **kwargs):
         """A reference implementation of a fitting function for a classifier.
 

bayesclass/tests/test_AODE.py

@@ -66,6 +66,7 @@ def test_AODE_states(clf, data):
     clf = AODE(random_state=17)
     clf.fit(*data)
     assert clf.states_ == 23
+    assert clf.depth_ == clf.states_
 
 
 def test_AODE_classifier(data, clf):

bayesclass/tests/test_KDB.py

@@ -58,6 +58,7 @@ def test_KDB_states(clf, data):
     clf = KDB(k=3, random_state=17)
     clf.fit(*data)
     assert clf.states_ == 23
+    assert clf.depth_ == clf.states_
 
 
 def test_KDB_classifier(data, clf):

bayesclass/tests/test_TAN.py

@@ -57,6 +57,7 @@ def test_TAN_states(clf, data):
     clf = TAN(random_state=17)
     clf.fit(*data)
     assert clf.states_ == 23
+    assert clf.depth_ == clf.states_
 
 
 def test_TAN_random_head(data):