edit test

ChinSekYi · Jul 7, 2024 · f265d35 · f265d35
1 parent a34f8b4
commit f265d35
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test.py b/test.py
@@ -29,7 +29,8 @@ def test_data_transformation_output():
 
     return train_arr, test_arr
 
-
+""""
+# takes a long time to run
 def test_model_output():
     obj = ModelTrainer()
     train_arr, test_arr = test_data_transformation_output()
@@ -39,6 +40,7 @@ def test_model_output():
     assert obj.model_trainer_config.trained_model_file_path == "artifacts/model.pkl"
     assert predicted.isnumeric()
     assert r2_score_value.isnumeric() and r2_score_value <= 1 and r2_score_value >= 0
+"""
 
 if __name__ == "__main__":
     test_model_output()