Skip to content

Commit

Permalink
typing in issue manager
Browse files Browse the repository at this point in the history
  • Loading branch information
FabianGroeger96 committed Apr 9, 2024
1 parent 5defc1e commit 464b9f1
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/cleaner/issue_manager.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from enum import Enum
from typing import Optional, Union

import numpy as np
import pandas as pd
from loguru import logger

Expand All @@ -24,7 +25,7 @@ def get_issues(
self,
issue_type: Union[str, IssueTypes],
return_as_df: bool = False,
):
) -> Union[np.ndarray, pd.DataFrame]:
if issue_type is type(IssueTypes):
issue_type = issue_type.value

Expand Down

0 comments on commit 464b9f1

Please sign in to comment.