Skip to content

Commit

Permalink
add data load and dump without parquet as a backup solution (mljar#308)
Browse files Browse the repository at this point in the history
  • Loading branch information
pplonski committed Apr 15, 2021
1 parent 27255b4 commit 95cc5b3
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions tests/tests_automl/test_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,3 +126,29 @@ def test_different_input_types(self):

self.assertIsInstance(pred_list, np.ndarray)
self.assertEqual(len(pred_list), X.shape[0])

def test_integration_float16_data(self):
a = AutoML(
results_path=self.automl_dir,
total_time_limit=1,
explain_level=0,
start_random_models=1,
)

X, y = datasets.make_classification(
n_samples=100,
n_features=5,
n_informative=4,
n_redundant=1,
n_classes=2,
n_clusters_per_class=3,
n_repeated=0,
shuffle=False,
random_state=0,
)
X = pd.DataFrame(X)
X = X.astype(np.float16)
a.fit(X, y)
p = a.predict(X)
self.assertIsInstance(p, np.ndarray)
self.assertEqual(len(p), X.shape[0])

0 comments on commit 95cc5b3

Please sign in to comment.