diff --git a/dask4dvc/__init__.py b/dask4dvc/__init__.py index b389df3..30ead2d 100644 --- a/dask4dvc/__init__.py +++ b/dask4dvc/__init__.py @@ -1,4 +1,5 @@ """The 'dask4dvc' package.""" + import importlib.metadata import logging import sys diff --git a/dask4dvc/cli/__init__.py b/dask4dvc/cli/__init__.py index 3620503..e650fcc 100644 --- a/dask4dvc/cli/__init__.py +++ b/dask4dvc/cli/__init__.py @@ -1,4 +1,5 @@ """The 'dask4dvc' CLI.""" + from dask4dvc.cli import main __all__ = ["main"] diff --git a/dask4dvc/dvc_repro.py b/dask4dvc/dvc_repro.py index 6cbde82..720f33c 100644 --- a/dask4dvc/dvc_repro.py +++ b/dask4dvc/dvc_repro.py @@ -1,4 +1,5 @@ """Dask4DVC to DVC repo interface.""" + import dataclasses import logging import subprocess @@ -161,7 +162,10 @@ def parallel_submit( client: dask.distributed.Client, repo: dvc.repo.Repo, stages: typing.Dict[PipelineStage, str], -) -> typing.Tuple[typing.Dict[PipelineStage, dask.distributed.Future], typing.List[str],]: +) -> typing.Tuple[ + typing.Dict[PipelineStage, dask.distributed.Future], + typing.List[str], +]: """Submit experiments in parallel.""" mapping = {} queue_entries = get_all_queue_entries(repo) diff --git a/dask4dvc/utils/dask.py b/dask4dvc/utils/dask.py index 7e391ce..3bfa598 100644 --- a/dask4dvc/utils/dask.py +++ b/dask4dvc/utils/dask.py @@ -1,4 +1,5 @@ """Utils that are related to 'dask'.""" + import logging import pathlib import typing diff --git a/tests/conftest.py b/tests/conftest.py index 4ae7c28..f9577fc 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,4 +1,5 @@ """global pytest fixtures.""" + import os import pathlib import shutil diff --git a/tests/test_dask4dvc.py b/tests/test_dask4dvc.py index 876e853..a79c121 100644 --- a/tests/test_dask4dvc.py +++ b/tests/test_dask4dvc.py @@ -1,4 +1,5 @@ """Test the 'dask4dvc' package.""" + import dask4dvc diff --git a/tests/test_repro.py b/tests/test_repro.py index 5731c83..ee4ee57 100644 --- a/tests/test_repro.py +++ b/tests/test_repro.py @@ -1,4 +1,5 @@ """Test the 'dask4dvc' CLI.""" + import pathlib import random diff --git a/tests/test_run.py b/tests/test_run.py index 81e054a..9ed8901 100644 --- a/tests/test_run.py +++ b/tests/test_run.py @@ -1,4 +1,5 @@ """Test the 'dask4dvc' CLI.""" + import pathlib import random import typing