Marked Resnet18 test as xfail

nod-ai · Dec 21, 2023 · ccaefaa · ccaefaa
1 parent 0fef610
commit ccaefaa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/examples/resnet-18/resnet-18.py b/examples/resnet-18/resnet-18.py
@@ -65,3 +65,6 @@ def compare_labels(id0, id1):
 y0 = shark_infer(x)
 y1 = forward(x)
 print_labels(y0)
+print(
+    f"Found {compare_labels(y0,y1)[0].size()[0]} discrepancies between turbine and standard result"
+)
diff --git a/python/turbine_models/tests/resnet_test.py b/python/turbine_models/tests/resnet_test.py
@@ -3,6 +3,7 @@
 from turbine_models.custom_models import resnet_18
 import unittest
 import os
+import pytest
 
 arguments = {
     "run_vmfb": True,
@@ -17,6 +18,9 @@
 
 
 class Resnet18Test(unittest.TestCase):
+    @pytest.mark.xfail(
+        reason="caused by lack of support for DenseResourceElementsAttr iteration over a generic FloatAttr"
+    )
     def testExportResnet18Model(self):
         with self.assertRaises(SystemExit) as cm:
             resnet_18.export_resnet_18_model(