fixing two typos in comments

tasks/common.py

@@ -53,7 +53,7 @@ class Task:
 
 class TaskMixture(Task):
     """
-    For SFT Training it becomes useful to train on a tax mixture of datasets.
+    For SFT Training it becomes useful to train on a mixture of datasets.
     Fun trick: if you wish to oversample any task, just pass it in multiple times in the list.
     """
 

tasks/gsm8k.py

@@ -74,7 +74,7 @@ class GSM8K(Task):
             else:
                 # Regular text in between tool calls
                 assistant_message_parts.append({"type": "text", "text": part})
-        # No put it all together
+        # Now put it all together
         messages = [
             {"role": "user", "content": question}, # note: simple string
             {"role": "assistant", "content": assistant_message_parts}, # note: list of parts (as dicts)