Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
AstrakhantsevaAA committed Jan 4, 2024
1 parent 0356072 commit 7f7c8bc
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
4 changes: 2 additions & 2 deletions dlt/cli/magics.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def display(self) -> t.Any:
)
return display # Assuming 'display' is a predefined callable

def success_message(self, message: str) -> HTML:
def success_message(self, message: str) -> t.Any:
msg = f'<div><span style="color: green; font-weight: bold">{message}</span></div>'
return self.display(HTML(msg))

Expand Down Expand Up @@ -181,7 +181,7 @@ def schema(self, line: str) -> t.Any:
return -1

@line_magic
def dlt_version(self, line) -> t.Any:
def dlt_version(self, line: str) -> t.Any:
"""
A dlt line magic command to display version information.
"""
Expand Down
5 changes: 4 additions & 1 deletion tests/cli/test_magics.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,10 @@ def test_init_command(shell_interactive):
@pytest.fixture
def run_pipeline():
load_info = dlt.pipeline(
pipeline_name="test_pipeline", destination="duckdb", dataset_name="mydata"
pipeline_name="test_pipeline",
destination="duckdb",
dataset_name="mydata",
full_refresh=True
).run([{"id": 1, "name": "John"}], table_name="users")
return load_info

Expand Down

0 comments on commit 7f7c8bc

Please sign in to comment.