X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;ds=sidebyside;f=tasks.py;h=78910a06d7a24977b462af0836a9219073381507;hb=b35745d09b33aed20670ecb96726f89206487a24;hp=181ac4475a2126df33206e2a4cea29a3f2d57833;hpb=732349f7c16e43ff84380d28e021d671f2c56492;p=picoclvr.git diff --git a/tasks.py b/tasks.py index 181ac44..78910a0 100755 --- a/tasks.py +++ b/tasks.py @@ -132,7 +132,7 @@ class TaskFromFile(Task): sequence = f.readline().strip() pred_mask = f.readline().strip() assert len(sequence) == len(pred_mask) - assert set(pred_mask) == {"0", "1", "2"}, f"{set(pred_mask)}" + assert set(pred_mask).issubset({"0", "1", "2"}), f"{set(pred_mask)}" pairs.append((sequence, pred_mask)) symbols = ["#"] + list(set("".join([x[0] for x in pairs])) - set(["#"]))