base: Correct parameter annotation

biolab · Jun 10, 2023 · b2c2761 · b2c2761
1 parent 0b8c286
commit b2c2761
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/orangecontrib/network/network/base.py b/orangecontrib/network/network/base.py
@@ -1,5 +1,5 @@
 from functools import reduce, wraps, partial
-from typing import Sequence
+from typing import Sequence, Union
 
 import numpy as np
 import scipy.sparse as sp
@@ -207,7 +207,11 @@ def wrapper(graph, *args, **kwargs):
 
 
 class Network:
-    def __init__(self, nodes: Sequence, edges: Sequence, name: str = "",
+    def __init__(self, nodes: Sequence,
+                 edges: Union[Edges,
+                              sp.csr_matrix,
+                              Sequence[Union[Edges, sp.csr_matrix]]],
+                 name: str = "",
                  coordinates: np.ndarray = None):
         """
         Attributes: