fix: [AnalystData] A quick and simple typing fix

MISP · Oct 30, 2024 · d5e472b · d5e472b
1 parent 7edb2e0
commit d5e472b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymisp/mispevent.py b/pymisp/mispevent.py
@@ -67,7 +67,7 @@ def add_note(self, note: str, language: str | None = None, **kwargs) -> MISPNote
         self.edited = True
         return the_note
 
-    def add_opinion(self, opinion: int, comment: str | None = None, **kwargs) -> MISPNote:  # type: ignore[no-untyped-def]
+    def add_opinion(self, opinion: int, comment: str | None = None, **kwargs) -> MISPOpinion:  # type: ignore[no-untyped-def]
         the_opinion = MISPOpinion()
         the_opinion.from_dict(opinion=opinion, comment=comment,
                               object_uuid=self.uuid, object_type=self.analyst_data_object_type,
@@ -76,7 +76,7 @@ def add_opinion(self, opinion: int, comment: str | None = None, **kwargs) -> MIS
         self.edited = True
         return the_opinion
 
-    def add_relationship(self, related_object_type: AbstractMISP | str, related_object_uuid: str | None, relationship_type: str, **kwargs) -> MISPNote:  # type: ignore[no-untyped-def]
+    def add_relationship(self, related_object_type: AbstractMISP | str, related_object_uuid: str | None, relationship_type: str, **kwargs) -> MISPRelationship:  # type: ignore[no-untyped-def]
         the_relationship = MISPRelationship()
         the_relationship.from_dict(related_object_type=related_object_type, related_object_uuid=related_object_uuid,
                                    relationship_type=relationship_type,