Fix the issue in test case

DataCanvasIO · Dec 7, 2020 · aa49056 · aa49056
1 parent 6fe4808
commit aa49056
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/models/hyper_dt_test.py b/tests/models/hyper_dt_test.py
@@ -81,7 +81,6 @@ def test_hyper_dt(self):
         y = np.random.randint(0, 2, size=(100), dtype='int')
         df = pd.DataFrame({'x1': x1, 'x2': x2, 'x3': x3, 'x4': x4})
         hdt.search(df, y, df, y, max_trails=3, epochs=1)
-        assert hdt.best_model
         best_trial = hdt.get_best_trail()
 
         estimator = hdt.final_train(best_trial.space_sample, df, y, epochs=1)