Bugfix: fixing CICD behave test (#1290)

featureform · Jan 25, 2024 · dbd3354 · dbd3354
1 parent 4f7a2da
commit dbd3354
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/client/src/featureform/enums.py b/client/src/featureform/enums.py
@@ -164,6 +164,7 @@ class ResourceType(Enum):
     # ResourceType is an enumeration representing the possible types of
     # resources that may be registered with Featureform. Each value is based
     # on OfflineResourceType in providers/offline.go
+
     NO_TYPE = 0
     LABEL = 1
     FEATURE = 2

diff --git a/tests/end_to_end/features/steps/serving.py b/tests/end_to_end/features/steps/serving.py
@@ -171,9 +171,9 @@ def step_impl(context):
     ]
     context.iter = context.client.batch_features(
         [
-            ("transaction_feature", context.variant),
-            ("balance_feature", context.variant),
-            ("perc_feature", context.variant),
+            ("transaction_feature", ff.get_run()),
+            ("balance_feature", ff.get_run()),
+            ("perc_feature", ff.get_run()),
         ]
     )