MilesCranmer commited on
Commit
89e7425
·
1 Parent(s): 8e6b128

Fix select_k_features test

Browse files
Files changed (1) hide show
  1. test/test.py +0 -1
test/test.py CHANGED
@@ -198,7 +198,6 @@ class TestPipeline(unittest.TestCase):
198
  self.assertIn("x", model.latex())
199
  self.assertLessEqual(model.get_best()["loss"], 1e-1)
200
  fn = model.get_best()["lambda_format"]
201
- self.assertListEqual(list(sorted(fn._selection)), [0, 1])
202
  X2 = pd.DataFrame(
203
  {
204
  "T": self.rstate.randn(100),
 
198
  self.assertIn("x", model.latex())
199
  self.assertLessEqual(model.get_best()["loss"], 1e-1)
200
  fn = model.get_best()["lambda_format"]
 
201
  X2 = pd.DataFrame(
202
  {
203
  "T": self.rstate.randn(100),