Adjust the number of features accordingly

HazyResearch · Sep 8, 2020 · aaa7598 · aaa7598
1 parent c619e3d
commit aaa7598
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/e2e/test_e2e.py b/tests/e2e/test_e2e.py
@@ -180,7 +180,7 @@ def test_e2e(database_session):
     featurizer.apply(split=1, train=True, parallelism=PARALLEL)
     assert session.query(Feature).count() == 214
     num_feature_keys = session.query(FeatureKey).count()
-    assert num_feature_keys == 1281
+    assert num_feature_keys == 1278
 
     # Test Dropping FeatureKey
     # Should force a row deletion

diff --git a/tests/e2e/test_incremental.py b/tests/e2e/test_incremental.py
@@ -181,7 +181,7 @@ def test_incremental(database_session):
     featurizer.update(new_docs, parallelism=PARALLEL)
     assert session.query(Feature).count() == len(train_cands[0])
     num_feature_keys = session.query(FeatureKey).count()
-    assert num_feature_keys == 2526
+    assert num_feature_keys == 2514
     F_train = featurizer.get_feature_matrices(train_cands)
     assert F_train[0].shape == (len(train_cands[0]), num_feature_keys)
     assert len(featurizer.get_keys()) == num_feature_keys