Skip to content

Commit 2fa31e0

Browse files
committed
#498: loosen type requirements in lbsObject
1 parent bb11351 commit 2fa31e0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/lbaf/Model/lbsObject.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
from .lbsBlock import Block
44
from .lbsObjectCommunicator import ObjectCommunicator
5+
from numbers import Number
56

67

78
class Object:
@@ -62,15 +63,15 @@ def __init__(
6263
if user_defined:
6364
# Object size is by definition its memory footprint
6465
if not isinstance((
65-
size := user_defined.get("task_footprint_bytes")), float) or size < 0.0:
66+
size := user_defined.get("task_footprint_bytes")), Number) or size < 0.0:
6667
raise TypeError(
6768
f"size: incorrect type {type(size)} or value: {size}")
6869
else:
6970
self.__size = size
7071

7172
# Object overhead is by definition its additional working memory
7273
if not isinstance((
73-
overhead := user_defined.get("task_working_bytes")), float) or overhead < 0.0:
74+
overhead := user_defined.get("task_working_bytes")), Number) or overhead < 0.0:
7475
raise TypeError(
7576
f"overhead: incorrect type {type(overhead)} or value: {overhead}")
7677
else:

0 commit comments

Comments
 (0)