handle dataframegroupby

src/vivarium/framework/results/interface.py

@@ -35,7 +35,7 @@
 ResultsFormatter = Callable[[str, pd.DataFrame], pd.DataFrame]
 """This is a Callable that takes a measure as a string and a DataFrame of observation results and returns formatted results."""
 ResultsGathererInput = Union[
-    pd.DataFrame, DataFrameGroupBy[tuple[str, ...] | str, bool], tuple[str, ...], None
+    pd.DataFrame, DataFrameGroupBy, tuple[str, ...], None  # type: ignore [type-arg]
 ]
 ResultsGatherer = Callable[[ResultsGathererInput], pd.DataFrame]
 """This is a Callable that optionally takes a possibly stratified population and returns new observation results."""