Merge nucleic/sleek-ember-seal-uady into dev

This commit is contained in:
2026-07-30 22:03:51 -07:00
parent 82545b833b
commit 929a873e95
3 changed files with 142 additions and 38 deletions
+31
View File
@@ -1,4 +1,5 @@
import argparse
import json
import sys
import tempfile
import unittest
@@ -94,6 +95,36 @@ class CoreMLQuantizationConfigTests(unittest.TestCase):
self.assertEqual([1.0], output["output"].tolist())
self.assertEqual([], list(root.glob("*.mlpackage")))
result_package = Path(tempfile.mkdtemp(suffix=".mlpackage"))
self.assertTrue(result_package.exists())
def test_activation_cache_requires_exact_contract(self):
expected = {
"schemaVersion": 1,
"sourcePackageSha256": "abc",
}
with tempfile.TemporaryDirectory() as temp:
root = Path(temp)
package = root / "a8.mlpackage"
package.mkdir()
manifest = root / "a8-manifest.json"
manifest.write_text(json.dumps(expected), encoding="utf-8")
self.assertTrue(
quantize_coreml._activation_cache_is_valid(
package,
manifest,
expected,
)
)
manifest.write_text("{}", encoding="utf-8")
self.assertFalse(
quantize_coreml._activation_cache_is_valid(
package,
manifest,
expected,
)
)
if __name__ == "__main__":
unittest.main()