Skip to content

Commit

Permalink
fixing lint on test file
Browse files Browse the repository at this point in the history
Signed-off-by: murilommen <[email protected]>
  • Loading branch information
murilommen committed Sep 20, 2022
1 parent 0dda2fe commit b7a15ad
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions plugins/flytekit-whylogs/tests/test_schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,11 @@

import pandas as pd
import whylogs as why
from flytekitplugins.whylogs.schema import WhylogsDatasetProfileTransformer
from whylogs.core import DatasetProfileView

from flytekitplugins.whylogs.schema import WhylogsDatasetProfileTransformer
from flytekit.core.context_manager import FlyteContextManager
from flytekit import task, workflow
from flytekit.core.context_manager import FlyteContextManager


@task
Expand Down Expand Up @@ -44,6 +44,6 @@ def test_to_html_method() -> None:
tf = WhylogsDatasetProfileTransformer()
profile_view = whylogs_profiling()
report = tf.to_html(FlyteContextManager.current_context(), profile_view, Type[DatasetProfileView])

assert isinstance(report, str)
assert "Profile Visualizer" in report

0 comments on commit b7a15ad

Please sign in to comment.