This is an automated email from the ASF dual-hosted git repository.
baunsgaard pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/systemds.git
The following commit(s) were added to refs/heads/master by this push:
new 60bd6d7 [MINOR] Disable Hyperband test if SYSTEMDS_ROOT not set
60bd6d7 is described below
commit 60bd6d7b956bebeadc092cd55211114d19b565f7
Author: baunsgaard <[email protected]>
AuthorDate: Wed Apr 28 13:54:02 2021 +0200
[MINOR] Disable Hyperband test if SYSTEMDS_ROOT not set
---
src/main/python/tests/frame/test_hyperband.py | 49 ++++++++++++++-------------
1 file changed, 26 insertions(+), 23 deletions(-)
diff --git a/src/main/python/tests/frame/test_hyperband.py
b/src/main/python/tests/frame/test_hyperband.py
index b470ed1..9874b89 100644
--- a/src/main/python/tests/frame/test_hyperband.py
+++ b/src/main/python/tests/frame/test_hyperband.py
@@ -57,29 +57,32 @@ class TestHyperband(unittest.TestCase):
pass
def test_hyperband(self):
- x_train = Matrix(self.sds, self.X_train)
- y_train = Matrix(self.sds, self.y_train)
- x_val = Matrix(self.sds, self.X_val)
- y_val = Matrix(self.sds, self.y_val)
- paramRanges = Matrix(self.sds, self.param_ranges)
- params = self.params
- [best_weights_mat, opt_hyper_params_df] = hyperband(
- X_train=x_train,
- y_train=y_train,
- X_val=x_val,
- y_val=y_val,
- params=params,
- paramRanges=paramRanges,
- ).compute()
- self.assertTrue(isinstance(best_weights_mat, np.ndarray))
- self.assertTrue(best_weights_mat.shape[0] == self.X_train.shape[1])
- self.assertTrue(best_weights_mat.shape[1] == self.y_train.shape[1])
-
- self.assertTrue(isinstance(opt_hyper_params_df, pd.DataFrame))
- self.assertTrue(opt_hyper_params_df.shape[0] == paramRanges.shape[0])
- self.assertTrue(opt_hyper_params_df.shape[1] == 1)
- for i, hyper_param in
enumerate(opt_hyper_params_df.values.flatten().tolist()):
- self.assertTrue(self.min_max_params[i][0] <= hyper_param <=
self.min_max_params[i][1])
+ if "SYSTEMDS_ROOT" in os.environ:
+ x_train = Matrix(self.sds, self.X_train)
+ y_train = Matrix(self.sds, self.y_train)
+ x_val = Matrix(self.sds, self.X_val)
+ y_val = Matrix(self.sds, self.y_val)
+ paramRanges = Matrix(self.sds, self.param_ranges)
+ params = self.params
+ [best_weights_mat, opt_hyper_params_df] = hyperband(
+ X_train=x_train,
+ y_train=y_train,
+ X_val=x_val,
+ y_val=y_val,
+ params=params,
+ paramRanges=paramRanges,
+ ).compute()
+ self.assertTrue(isinstance(best_weights_mat, np.ndarray))
+ self.assertTrue(best_weights_mat.shape[0] == self.X_train.shape[1])
+ self.assertTrue(best_weights_mat.shape[1] == self.y_train.shape[1])
+
+ self.assertTrue(isinstance(opt_hyper_params_df, pd.DataFrame))
+ self.assertTrue(opt_hyper_params_df.shape[0] ==
paramRanges.shape[0])
+ self.assertTrue(opt_hyper_params_df.shape[1] == 1)
+ for i, hyper_param in
enumerate(opt_hyper_params_df.values.flatten().tolist()):
+ self.assertTrue(self.min_max_params[i][0] <= hyper_param <=
self.min_max_params[i][1])
+ else:
+ print("to enable hyperband tests, set SYSTEMDS_ROOT")
if __name__ == "__main__":
