Skip to content
This repository has been archived by the owner on Jul 18, 2024. It is now read-only.

Commit

Permalink
bump version
Browse files Browse the repository at this point in the history
  • Loading branch information
PythonFZ committed Apr 13, 2023
1 parent 43b1224 commit c15294b
Show file tree
Hide file tree
Showing 5 changed files with 10 additions and 8 deletions.
4 changes: 0 additions & 4 deletions dask4dvc.yaml

This file was deleted.

7 changes: 5 additions & 2 deletions dask4dvc/utils/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,11 +9,14 @@


def timeit(func: typing.Callable) -> typing.Callable:
"""Timeit decorator for benchmarking."""
"""Timeit decorator for benchmarking.
# TODO remove and replace with proper benchmarking / profiling
"""

@functools.wraps(func)
def wrapper(*args: tuple, **kwargs: dict) -> typing.Any:
"""Wrapper for timeit."""
"""Wrap to time the function."""
start_time = time.time()
result = func(*args, **kwargs)
stop_time = time.time()
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "dask4dvc"
version = "0.1.2"
version = "0.1.3"
description = "Use dask to run the DVC graph"
authors = ["zincwarecode <[email protected]>"]
license = "Apache-2.0"
Expand Down
2 changes: 1 addition & 1 deletion tests/test_dask4dvc.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,4 +4,4 @@

def test_version() -> None:
"""Test Version."""
assert dask4dvc.__version__ == "0.1.2"
assert dask4dvc.__version__ == "0.1.3"
3 changes: 3 additions & 0 deletions tests/test_utils_dvc.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Test the 'dask4dvc' CLI."""
import os
import pathlib

import dask_jobqueue
Expand Down Expand Up @@ -30,6 +31,8 @@ def test_repro(single_node_repo: tuple, targets: list, options: list) -> None:

def test_dask_get_cluster_from_config(tmp_path: pathlib.Path) -> None:
"""Test 'dask4dvc.utils.dask.get_cluster_from_config'."""
os.chdir(tmp_path)

file = pathlib.Path("dask4dvc.yaml")
config = {"default": {"PBSCluster": {"cores": 1, "memory": "1GB"}}}
file.write_text(yaml.safe_dump(config))
Expand Down

0 comments on commit c15294b

Please sign in to comment.