Import pytest in failed test files

Signed-off-by: Cody Lin <codyl@twitter.com>
feast-dev · Jun 20, 2021 · 4ec510d · 4ec510d
1 parent 5abc428
commit 4ec510d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/sdk/python/tests/test_cli_local.py b/sdk/python/tests/test_cli_local.py
@@ -4,6 +4,7 @@
 from textwrap import dedent
 
 import assertpy
+import pytest
 
 from feast.feature_store import FeatureStore
 from tests.cli_utils import CliRunner

diff --git a/sdk/python/tests/test_partial_apply.py b/sdk/python/tests/test_partial_apply.py
@@ -1,3 +1,4 @@
+import pytest
 from google.protobuf.duration_pb2 import Duration
 
 from feast import BigQuerySource, Feature, FeatureView, ValueType