Resolve "integration-sota"

Closes #15 (closed)

code/bolsonaro/models/kmeans_forest_regressor.py

@@ -121,15 +121,4 @@ class KMeansForestClassifier(KmeansForest, metaclass=ABCMeta):
     def _get_best_tree_index(self, y_preds, y_true):
         score = self._score_metric(y_preds, y_true)
         best_tree_index = np.argmax(score)  # get best scoring tree (the one with lowest mse)
-        return best_tree_index
-
-if __name__ == "__main__":
-    from sklearn import datasets
-    from bolsonaro.models.model_parameters import ModelParameters
-
-    X, y = datasets.fetch_california_housing(return_X_y=True)
-    ModelParameters(extracted_forest_size=100,
-                    normalize_D=True,
-                    )
-
-    k_reg = KMeansForestRegressor()
\ No newline at end of file
+        return best_tree_index
\ No newline at end of file