@@ -81,16 +81,6 @@ def test_reqs_on_disk(self, experiment_run, model_for_deployment, output_path):
8181 with open (requirements_file , 'r' ) as f :
8282 assert set (f .read ().split ()) <= set (retrieved_requirements .split ())
8383
84- def test_with_data (self , experiment_run , model_for_deployment ):
85- """`train_features` and `train_targets` are joined into a single CSV"""
86- experiment_run .log_model_for_deployment (** model_for_deployment )
87-
88- data_csv = experiment_run .get_artifact ("train_data" ).read ()
89-
90- X_train = model_for_deployment ['train_features' ]
91- y_train = model_for_deployment ['train_targets' ]
92- assert X_train .join (y_train ).to_csv (index = False ) == six .ensure_str (data_csv )
93-
9484
9585@pytest .mark .not_oss
9686class TestLogModel :
@@ -563,22 +553,18 @@ def test_series(self, experiment_run, model_for_deployment):
563553 X_train = model_for_deployment ['train_features' ]
564554 y_train = model_for_deployment ['train_targets' ]
565555
556+ # no errors
566557 experiment_run .log_training_data (X_train , y_train )
567558
568- data_csv = experiment_run .get_artifact ("train_data" ).read ()
569- assert X_train .join (y_train ).to_csv (index = False ) == six .ensure_str (data_csv )
570-
571559 def test_dataframe (self , experiment_run , model_for_deployment ):
572560 X_train = model_for_deployment ['train_features' ]
573561 y_train = model_for_deployment ['train_targets' ]
574562
575563 y_train = y_train .to_frame ()
576564
565+ # no errors
577566 experiment_run .log_training_data (X_train , y_train )
578567
579- data_csv = experiment_run .get_artifact ("train_data" ).read ()
580- assert X_train .join (y_train ).to_csv (index = False ) == six .ensure_str (data_csv )
581-
582568
583569class TestHistogram :
584570 @staticmethod
0 commit comments