fix: integ test and black check

aws · Mar 14, 2024 · e68d6e5 · e68d6e5
1 parent e747b44
commit e68d6e5
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/tests/integ/core/__init__.py b/tests/integ/core/__init__.py
@@ -10,4 +10,4 @@
 # distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
-from __future__ import absolute_import
+from __future__ import absolute_import
diff --git a/tests/integ/core/conftest.py b/tests/integ/core/conftest.py
@@ -15,7 +15,14 @@
 import pytest
 import pathlib
 
+
+def pytest_configure(config):
+    config.addinivalue_line(
+        "markers", "core: subset of integ tests that must run for every pull request"
+    )
+
+
 def pytest_collection_modifyitems(config, items):
     for item in items:
-         if 'tests/integ/core' in str(item.fspath):
+        if "tests/integ/core" in str(item.fspath):
             item.add_marker(pytest.mark.core)
diff --git a/tests/integ/core/test_tf.py b/tests/integ/core/test_tf.py
@@ -32,7 +32,7 @@
 
 ROLE = "SageMakerRole"
 
-RESOURCE_PATH = os.path.join(os.path.dirname(__file__), "..", "data")
+RESOURCE_PATH = os.path.join(os.path.dirname(__file__), "..", "..", "data")
 MNIST_RESOURCE_PATH = os.path.join(RESOURCE_PATH, "tensorflow_mnist")
 TFS_RESOURCE_PATH = os.path.join(RESOURCE_PATH, "tfs", "tfs-test-entrypoint-with-handler")