lint fix and inference context fix

argmaxinc · Dec 4, 2024 · e56cd0a · e56cd0a
1 parent 0653285
commit e56cd0a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/test_evaluate.py b/tests/test_evaluate.py
@@ -61,6 +61,12 @@ def setUpClass(cls) -> None:
             shell=True
         ).stdout.decode('utf-8').strip()[:7]
 
+        inference_context_spec_dict = None
+        try:
+            inference_context_spec_dict = cls.inference_context.spec_dict()
+        except Exception as e:
+            logger.warning(f"Inference context spec dict failed: {e}")
+
         cls.results = {
             "results": evaluate(
                 cls.pipeline,
@@ -76,7 +82,7 @@ def setUpClass(cls) -> None:
                 "dataset_name": TEST_DATASET_NAME,
                 "model_version": TEST_MODEL_VERSION,
                 "whisperkittools_commit_hash": wkt_commit_hash,
-                "inference_context": cls.inference_context.spec_dict(),
+                "inference_context": inference_context_spec_dict,
                 "model_repo_id": MODEL_REPO_ID
             }
         }

diff --git a/whisperkit/android/utils.py b/whisperkit/android/utils.py
@@ -20,7 +20,7 @@
     "tiny", "tiny.en", "base", "base.en", "small", "small.en"
 ]
 
-# Reference: 
+# Reference:
 # https://app.aihub.qualcomm.com/docs/hub/compile_examples.html#compiling-pytorch-model-to-a-qnn-model-library
 
 TARGET_TABLE = {