return

Author: jmafoster1

causal_testing/testing/causal_test_adequacy.py

@@ -55,4 +55,4 @@ def measure_adequacy_k_folds(self, k: int = 10, random_state=0):
             estimator.df = estimator.df.iloc[train_inx]
             test_result = estimator.model.predict(test)
             results.append(np.sqrt(mse(test_result, test[self.test_case.base_test_case.outcome_variable.name])).mean())
-        print("K-score", np.mean(results))
+        return np.mean(results)