auto linting

serengil · Dec 23, 2023 · fd0d0dd · fd0d0dd
1 parent dedfb28
commit fd0d0dd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/chefboost/training/Training.py b/chefboost/training/Training.py
@@ -511,12 +511,12 @@ def buildDecisionTree(
 
     if df.Decision.dtypes == "object":  # classification
         pivot = pd.DataFrame(subdataset.Decision.value_counts()).reset_index()
-        
+
         if pd.__version__.split(".")[0] == "1":
             pivot = pivot.rename(columns={"Decision": "Instances", "index": "Decision"})
-        else: # if pd.__version__.split(".")[0] == "2":
+        else:  # if pd.__version__.split(".")[0] == "2":
             pivot = pivot.rename(columns={"Decision": "Instances", "count": "Decision"})
-        
+
         pivot = pivot.sort_values(by=["Instances"], ascending=False).reset_index()
 
         else_decision = f"return '{pivot.iloc[0].Decision}'"