add dataset updates with doc updates

tests/test_Dataset.py

@@ -1,6 +1,8 @@
 import unittest
+from typing import cast
 
 from server.Dataset import Dataset
+from server.GameBoard import GameBoard
 
 class DummySnake:
   def get_history(self):
@@ -25,7 +27,7 @@ class DummyGameBoard:
 
 class TestDataset(unittest.TestCase):
   def test_build_only_good_moves_for_wins(self):
-    dataset = Dataset(DummyGameBoard(["me"]))
+    dataset = Dataset(cast(GameBoard, DummyGameBoard(["me"])))
     payload = dataset.build(only_good_moves=True)
 
     self.assertTrue(payload["did_win"])
@@ -33,15 +35,15 @@ class TestDataset(unittest.TestCase):
     self.assertTrue(all(sample["is_good_move"] for sample in payload["samples"]))
 
   def test_build_returns_no_samples_for_losses_when_only_good(self):
-    dataset = Dataset(DummyGameBoard(["enemy"]))
+    dataset = Dataset(cast(GameBoard, DummyGameBoard(["enemy"])))
     payload = dataset.build(only_good_moves=True)
 
     self.assertFalse(payload["did_win"])
     self.assertEqual(payload["total_samples"], 0)
 
   def test_labels_by_turn(self):
-    winner_labels = Dataset(DummyGameBoard(["me"])).labels_by_turn()
-    loser_labels = Dataset(DummyGameBoard(["enemy"])).labels_by_turn()
+    winner_labels = Dataset(cast(GameBoard, DummyGameBoard(["me"]))).labels_by_turn()
+    loser_labels = Dataset(cast(GameBoard, DummyGameBoard(["enemy"]))).labels_by_turn()
 
     self.assertEqual(winner_labels, {1: True, 2: True})
     self.assertEqual(loser_labels, {1: False, 2: False})