fix the type hint error for | in Python3.7 by using typing.Union

uber · Apr 13, 2024 · 6fd5df2 · 6fd5df2
1 parent 8528f35
commit 6fd5df2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/causalml/inference/tree/plot.py b/causalml/inference/tree/plot.py
@@ -4,6 +4,7 @@
 """
 
 from collections import defaultdict
+from typing import Union
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -399,7 +400,7 @@ def __init__(
         self.treatment_groups = treatment_groups
 
     def node_to_str(
-        self, tree: _tree.Tree, node_id: int, criterion: str | object
+        self, tree: _tree.Tree, node_id: int, criterion: Union[str, object]
     ) -> str:
         """
         Generate the node content string