fix: Fix Feast UI failure with new way of specifying entities

adchia · adchia · commit 956c833269c4 · 2022-06-08T22:05:59.000-04:00
Signed-off-by: Danny Chiao &lt;danny@tecton.ai&gt;
diff --git a/ui/feature_repo/test_get_features.py b/ui/feature_repo/test_get_features.py
@@ -1,12 +1,11 @@
 import pandas as pd
-from feast import FeatureStore
-from feast.infra.offline_stores.file import SavedDatasetFileStorage
-
-from feast.dqm.profilers.ge_profiler import ge_profiler
-
 from great_expectations.core.expectation_suite import ExpectationSuite
 from great_expectations.dataset import PandasDataset
 
+from feast import FeatureStore
+from feast.dqm.profilers.ge_profiler import ge_profiler
+from feast.infra.offline_stores.file import SavedDatasetFileStorage
+
 DELTA = 0.1  # controlling allowed window in fraction of the value on scale [0, 1]
 # Note: the GE integration allows asserting differences between datasets. The "ds" below is the reference dataset to check and this generates the expectation suite which can be used against future datasets.
 # It's used via ge.validate(new_dataset, ExpectationSuite)
