small changes

.github/workflows/ci_graviton_cpu.yml

@@ -17,11 +17,7 @@ concurrency:
 
 jobs:
   ci-cpu:
-    runs-on: ${{ matrix.os }}
-    strategy:
-      fail-fast: false
-      matrix:
-        os: [graviton-test]
+    runs-on: [self-hosted, graviton-test]
     steps:
       - name: Setup Python
         uses: actions/setup-python@v5

ts/torch_handler/unit_tests/test_object_detector.py

@@ -5,6 +5,7 @@
 Ensures it can load and execute an example model
 """
 
+import platform
 import sys
 from pathlib import Path
 
@@ -72,7 +73,9 @@ def handler(context):
     return handler
 
 
-@pytest.mark.skip(reason="Skipping this test for now")
+@pytest.mark.skipif(
+    platform.machine() == "aarch64", reason="Test skipped on aarch64 architecture"
+)
 def test_handle(handler, context, image_bytes):
     test_data = [{"data": image_bytes}] * 2
     results = handler.handle(test_data, context)