updated to Ray 2.7

2023-09-18 22:03:20 -07:00
parent 71b3d50a05
commit b98bd5b1ae
15 changed files with 3484 additions and 2086 deletions
--- a/tests/code/test_data.py
+++ b/tests/code/test_data.py
@@ -54,5 +54,7 @@ def test_preprocess(df, class_to_index):

 def test_fit_transform(dataset_loc, preprocessor):
    ds = data.load_data(dataset_loc=dataset_loc)
-    preprocessor.fit_transform(ds)
+    preprocessor = preprocessor.fit(ds)
+    preprocessed_ds = preprocessor.transform(ds)
    assert len(preprocessor.class_to_index) == 4
+    assert ds.count() == preprocessed_ds.count()