@@ -81,16 +81,6 @@ def test_reqs_on_disk(self, experiment_run, model_for_deployment, output_path):
81
81
with open (requirements_file , 'r' ) as f :
82
82
assert set (f .read ().split ()) <= set (retrieved_requirements .split ())
83
83
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
-
94
84
95
85
@pytest .mark .not_oss
96
86
class TestLogModel :
@@ -563,22 +553,18 @@ def test_series(self, experiment_run, model_for_deployment):
563
553
X_train = model_for_deployment ['train_features' ]
564
554
y_train = model_for_deployment ['train_targets' ]
565
555
556
+ # no errors
566
557
experiment_run .log_training_data (X_train , y_train )
567
558
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
-
571
559
def test_dataframe (self , experiment_run , model_for_deployment ):
572
560
X_train = model_for_deployment ['train_features' ]
573
561
y_train = model_for_deployment ['train_targets' ]
574
562
575
563
y_train = y_train .to_frame ()
576
564
565
+ # no errors
577
566
experiment_run .log_training_data (X_train , y_train )
578
567
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
-
582
568
583
569
class TestHistogram :
584
570
@staticmethod
0 commit comments