Skip to content

Commit

Permalink
fix marker type hints missing
Browse files Browse the repository at this point in the history
  • Loading branch information
pedrolabonia committed May 4, 2022
1 parent b4b5256 commit 872279e
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions pydavinci/wrappers/marker.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,14 +73,14 @@ def __init__(self, obj: "PydavinciParent") -> None:
def add(
self,
frameid: int,
color: str,
color: COLORS,
name: str,
*,
note: str = "",
duration: int = 1,
customdata: str = "",
overwrite: bool = False,
) -> Optional["Marker"]:
) -> "Marker":
"""
Adds a marker.
Expand All @@ -106,7 +106,7 @@ def add(
log.info(
f"Marker at {frameid} already exists. Skipping... If you want to overwrite, use overwrite = True"
)
return None
return None # type: ignore
else:
log.warn(f"Marker at frame {frameid} already exists. Overwriting ...")
f = frameid
Expand All @@ -131,7 +131,7 @@ def add(
log.error(
"Couldn't add marker. Make sure the frameid is correct and the duration isn't bigger than the clips' length."
)
return None
return None # type: ignore

def find(self, needle: str) -> Optional["Marker"]:
"""Finds the first marker that matches `needle` for the `Marker's` `note`, `name` or `customdata`
Expand Down Expand Up @@ -176,7 +176,7 @@ def delete(
self,
*,
frameid: int = 0,
color: str = "",
color: COLORS = "", # type: ignore
customdata: str = "",
) -> bool:
"""
Expand Down Expand Up @@ -376,7 +376,7 @@ def _update(self, key: ATTRS, value: Union[str, int]) -> None:
self._data[key] = value # type: ignore
self._interface.add(
frameid=self._data["frameid"],
color=self._data["color"],
color=self._data["color"], # type: ignore
name=self._data["name"],
note=self._data["note"],
duration=self._data["duration"],
Expand Down

0 comments on commit 872279e

Please sign in to comment.