fix typing

bdpedigo · Mar 11, 2024 · 5c1b665 · 5c1b665
1 parent 9982ef0
commit 5c1b665
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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: