Update TODOs

src/ashpy/contexts/context.py

@@ -55,7 +55,6 @@ def __init__(
         """
         self._distribute_strategy = tf.distribute.get_strategy()
 
-        # TODO: are metrics really needed right now?
         self._metrics = metrics if metrics else ()
         self._dataset = dataset
         self._log_eval_mode = log_eval_mode

src/ashpy/restorers/restorer.py

@@ -120,6 +120,7 @@ def _check_model_construction(restored_model: tf.keras.Model) -> bool:
         as layers.weights.
 
         TODO: add docs for the exception.
+        TODO: add test case for the Sequential without input shape
         """
         try:
             if restored_model.weights == []:

tests/callbacks/__init__.py

@@ -1 +1,6 @@
-"""Test for callbacks."""
+"""
+Test for callbacks.
+
+TODO: test a callback that retrieves a Metric from its Context
+TODO: Restructure these tests
+"""