add comment for posterity

vanvalenlab · Jan 15, 2021 · feef87b · feef87b
1 parent 3c050d2
commit feef87b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepcell_tracking/tracking_test.py b/deepcell_tracking/tracking_test.py
@@ -72,6 +72,7 @@ class DummyModel(object):  # pylint: disable=useless-object-inheritance
 
     def predict(self, data):
         if isinstance(data, dict):
+            # Grab a random value from the data dict and select batch dim
             batches = 0 if not data else next(iter(data.values())).shape[0]
         else:
             batches = next(iter(data.values())).shape[0]