@@ -433,7 +433,7 @@ def __init__(
433
433
cls_name : str ,
434
434
direction : int ,
435
435
manager : type [AsyncRelationshipManager ] = AsyncRelationshipManager ,
436
- model : Optional [AsyncStructuredRel ] = None ,
436
+ model : Optional [type [ AsyncStructuredRel ] ] = None ,
437
437
) -> None :
438
438
self ._validate_class (cls_name , model )
439
439
@@ -486,7 +486,7 @@ def __init__(
486
486
adb ._NODE_CLASS_REGISTRY [label_set ] = model
487
487
488
488
def _validate_class (
489
- self , cls_name : str , model : Optional [AsyncStructuredRel ] = None
489
+ self , cls_name : str , model : Optional [type [ AsyncStructuredRel ] ] = None
490
490
) -> None :
491
491
if not isinstance (cls_name , (str , object )):
492
492
raise ValueError ("Expected class name or class got " + repr (cls_name ))
@@ -552,7 +552,7 @@ def __init__(
552
552
cls_name : str ,
553
553
relation_type : str ,
554
554
cardinality : type [AsyncRelationshipManager ] = AsyncZeroOrMore ,
555
- model : Optional [AsyncStructuredRel ] = None ,
555
+ model : Optional [type [ AsyncStructuredRel ] ] = None ,
556
556
) -> None :
557
557
super ().__init__ (
558
558
relation_type , cls_name , OUTGOING , manager = cardinality , model = model
@@ -565,7 +565,7 @@ def __init__(
565
565
cls_name : str ,
566
566
relation_type : str ,
567
567
cardinality : type [AsyncRelationshipManager ] = AsyncZeroOrMore ,
568
- model : Optional [AsyncStructuredRel ] = None ,
568
+ model : Optional [type [ AsyncStructuredRel ] ] = None ,
569
569
) -> None :
570
570
super ().__init__ (
571
571
relation_type , cls_name , INCOMING , manager = cardinality , model = model
@@ -578,7 +578,7 @@ def __init__(
578
578
cls_name : str ,
579
579
relation_type : str ,
580
580
cardinality : type [AsyncRelationshipManager ] = AsyncZeroOrMore ,
581
- model : Optional [AsyncStructuredRel ] = None ,
581
+ model : Optional [type [ AsyncStructuredRel ] ] = None ,
582
582
) -> None :
583
583
super ().__init__ (
584
584
relation_type , cls_name , EITHER , manager = cardinality , model = model
0 commit comments