Skip to content

Commit ea28c7e

Browse files
committed
mypy
1 parent a55f3d5 commit ea28c7e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

onnx_diagnostic/helpers/log_helper.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1658,7 +1658,7 @@ def unbiased_export(df):
16581658

16591659
def view(
16601660
self,
1661-
view_def: Union[str, CubeViewDef],
1661+
view_def: Optional[Union[str, CubeViewDef]],
16621662
return_view_def: bool = False,
16631663
verbose: int = 0,
16641664
) -> Union[
@@ -1672,7 +1672,8 @@ def view(
16721672
:param view_def: view definition or a string
16731673
:param return_view_def: returns the view definition as well
16741674
:param verbose: verbosity level
1675-
:return: dataframe
1675+
:return: dataframe or a couple (dataframe, view definition),
1676+
both of them can be one if view_def cannot be interpreted
16761677
"""
16771678
assert view_def is not None, "view_def is None, this is not allowed."
16781679
if isinstance(view_def, str):

0 commit comments

Comments
 (0)