From 5c1b665d47c9a05d837966ee9d0f07c6ec69071a Mon Sep 17 00:00:00 2001 From: Ben Pedigo Date: Mon, 11 Mar 2024 13:30:40 -0700 Subject: [PATCH] fix typing --- networkframe/groupby.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/networkframe/groupby.py b/networkframe/groupby.py index c68491b..e476110 100644 --- a/networkframe/groupby.py +++ b/networkframe/groupby.py @@ -1,3 +1,5 @@ +from typing import Union + class NodeGroupBy: """A class for grouping a `NetworkFrame` by a set of labels.""" @@ -20,6 +22,7 @@ def __init__(self, frame, source_groupby, target_groupby, induced: bool = False) self._target_groupby = target_groupby self.induced = induced + self._axis: Union[str, int] if source_groupby is None: self._axis = 1 elif target_groupby is None: