Both Unsloth DeepSeek sets are offered on a DGX Spark: both passed a real Spark run (AARCH64_NOT_YET is empty); tests keep the mechanism

This commit is contained in:
Niko1221 committed 2026-10-30 22:50:34 +01:00
1 parent 1a26492064
commit 38f56619cc
2 files changed
+17 -3

No files matched your search

+3 -2
View File
@@ -618,8 +618,9 @@ MAC_SET_PACK = {"IST-DAS-2.00bpw": "IQ2_XXS", "IST-DAS-2.25bpw": "IQ2_XS", "UD-I
# Remove a name here when its set has run on a Mac.
MAC_NOT_YET = ("UD-IQ1_M",)
MAC_NOT_YET_LINE = "Unsloth's UD-IQ1_M set doesn't run on a Mac yet; the UD-IQ1_S, 2.00 and 2.25 sets do."
# The same two sets were never run on a DGX Spark (linux-aarch64, GB10) either: kept off it the same way until a Spark run passes (remove the names then).
AARCH64_NOT_YET = ("UD-IQ1_S", "UD-IQ1_M")
# The same two sets were kept off a DGX Spark (linux-aarch64, GB10) until a Spark run passed: both passed there on 2026-10-30 (real `strata launch --model-path`,
# 10 of 10 answers, tool call correct), so the tuple is empty. Put a name back here to hide a set from the Spark again.
AARCH64_NOT_YET = ()
AARCH64_NOT_YET_LINE = "Unsloth's 1-bit DeepSeek sets were not tested on a DGX Spark yet; the 2.00 and 2.25 sets were."
+14 -1
View File
@@ -22,7 +22,7 @@ MAC = {**HW_12, "mac": True, "ram_gb": 128.0, "disk_free_gb": 900.0, "gpus": [],
PC = {**HW_12, "ram_gb": 64.0, "disk_free_gb": 900.0, "engine_vendor": "nvidia"}
UD = ("UD-IQ1_M",) # the sets kept off a Mac
RUNS = "UD-IQ1_S" # the one that was run there
BOTH = ("UD-IQ1_S", "UD-IQ1_M") # both are still kept off a DGX Spark
BOTH = ("UD-IQ1_S", "UD-IQ1_M") # both ran on a DGX Spark (10-30): AARCH64_NOT_YET is empty; SparkGate patches it back to test the mechanism
def ud_files(model):
@@ -213,6 +213,9 @@ class SparkGate(unittest.TestCase):
p = mock.patch.object(Catalog, "hardware", lambda s, fresh=False: dict(self.hw))
p.start()
self.addCleanup(p.stop)
q = mock.patch.object(dsv41_fit, "AARCH64_NOT_YET", BOTH) # the mechanism, as it was while the sets were kept off
q.start()
self.addCleanup(q.stop)
def test_the_rule(self):
self.assertEqual(dsv41_fit.AARCH64_NOT_YET, BOTH)
@@ -272,5 +275,15 @@ class SparkGate(unittest.TestCase):
self.assertIn("nothing has been downloaded", text)
class SparkOffered(unittest.TestCase):
"""Both Unsloth sets ran on a DGX Spark (2026-10-30): the production rule hides nothing there."""
def test_nothing_is_hidden_on_a_spark(self):
self.assertEqual(dsv41_fit.AARCH64_NOT_YET, ())
with spark():
for m in BOTH + ("IST-DAS-2.25bpw",):
self.assertIsNone(dsv41_fit.not_here(m))
if __name__ == "__main__":
unittest.main()