Minor typing

DIAGNijmegen · Aug 27, 2024 · ffbb9a9 · ffbb9a9
1 parent 97d74c6
commit ffbb9a9
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/gcapi/models_proto.py b/gcapi/models_proto.py
@@ -13,7 +13,7 @@
     HyperlinkedComponentInterfaceValue,
     HyperlinkedImage,
 )
-from gcapi.typing import CIVSetDescription, FileSource
+from gcapi.typing import CIVSet, CIVSetDescription, FileSource
 
 
 class TooManyFiles(ValueError):
@@ -96,9 +96,7 @@ def __init__(
             FileSource, HyperlinkedComponentInterfaceValue, HyperlinkedImage
         ],
         interface: ComponentInterface,
-        parent: Optional[
-            Union[ArchiveItem, ArchiveItemPost, DisplaySet, DisplaySetPost]
-        ] = None,
+        parent: Optional[CIVSet] = None,
         **kwargs,
     ):
         super().__init__(**kwargs)