Skip to content

Commit f0dd7c5

Browse files
committed
fixed typing in bvh
1 parent f5208e1 commit f0dd7c5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/compas_model/models/bvh.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from typing import TYPE_CHECKING
2+
from typing import Iterable
23
from typing import Optional
34
from typing import Type
45
from typing import Union
@@ -36,7 +37,7 @@ def compute_box(self) -> Box:
3637
class ElementBVH(BVH):
3738
def __init__(
3839
self,
39-
nodetype: Optional[Union[ElementAABBNode, ElementOBBNode]] = ElementAABBNode,
40+
nodetype: Optional[Union[Type[ElementAABBNode], Type[ElementOBBNode]]] = ElementAABBNode,
4041
max_depth=None,
4142
leafsize=1,
4243
**kwargs,
@@ -46,8 +47,8 @@ def __init__(
4647
@classmethod
4748
def from_elements(
4849
cls,
49-
elements: list["Element"],
50-
nodetype: Optional[Union[Type[ElementAABBNode], Type[ElementOBBNode]]] = AABBNode,
50+
elements: Iterable["Element"],
51+
nodetype: Optional[Union[Type[ElementAABBNode], Type[ElementOBBNode]]] = ElementAABBNode,
5152
max_depth: Optional[int] = None,
5253
leafsize: int = 1,
5354
) -> "ElementBVH":

0 commit comments

Comments
 (0)