Fix usage test

Signed-off-by: Felix Wang <wangfelix98@gmail.com>
feast-dev · Jan 24, 2022 · 2e8961f · 2e8961f
1 parent 8e0b5dd
commit 2e8961f
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/sdk/python/tests/integration/e2e/test_usage_e2e.py b/sdk/python/tests/integration/e2e/test_usage_e2e.py
@@ -66,16 +66,10 @@ def test_usage_on(dummy_exporter, enabling_toggle):
 
         test_feature_store.apply([entity])
 
-        assert len(dummy_exporter) == 3
-        assert {
-            "entrypoint": "feast.infra.local.LocalRegistryStore.get_registry_proto"
-        }.items() <= dummy_exporter[0].items()
-        assert {
-            "entrypoint": "feast.infra.local.LocalRegistryStore.update_registry_proto"
-        }.items() <= dummy_exporter[1].items()
+        assert len(dummy_exporter) == 1
         assert {
             "entrypoint": "feast.feature_store.FeatureStore.apply"
-        }.items() <= dummy_exporter[2].items()
+        }.items() <= dummy_exporter[0].items()
 
 
 @pytest.mark.integration