Add version cap (#65)

* Add version cap

* fix

.pre-commit-config.yaml

@@ -20,7 +20,7 @@ repos:
           - id: end-of-file-fixer
           - id: trailing-whitespace
           - id: check-vcs-permalinks
-    - repo: https://gitlab.com/pycqa/flake8
+    - repo: https://github.com/pycqa/flake8
       rev: '3.8.3'
       hooks:
           - id: flake8

RELEASE.md

@@ -1,5 +1,9 @@
 # Release Notes
 
+## 0.1.3
+
+-  Added Fugue version constraint to avoid breaking changes
+
 ## 0.1.2
 
 -  [54](https://github.com/fugue-project/tune/pull/54): Integrated with MLFlow

setup.py

@@ -30,7 +30,7 @@ def get_version() -> str:
     author_email="goodwanghan@gmail.com",
     keywords="hyper parameter hyperparameter tuning tune tuner optimzation",
     url="http://github.com/fugue-project/tune",
-    install_requires=["fugue>=0.7.0", "cloudpickle"],
+    install_requires=["fugue>=0.7.0,<=0.8.1", "cloudpickle"],
     extras_require={
         "hyperopt": ["hyperopt"],
         "optuna": ["optuna"],

tests/tune/concepts/test_dataset.py

@@ -37,6 +37,7 @@ def assert_count(df: DataFrame, n: int, schema=None) -> None:
     with FugueWorkflow() as dag:
         df = builder.build(dag).data
         df.show()
+    dag.run()
 
     df1 = ArrayDataFrame([[0, 1], [1, 1], [0, 2]], "a:int,b:int")
 
@@ -52,6 +53,7 @@ def assert_count(df: DataFrame, n: int, schema=None) -> None:
             assert_count,
             params=dict(n=2, schema=f"__tune_df__x:str,{TUNE_DATASET_TRIALS}:str"),
         )
+    dag.run()
 
     space = Space(b=Rand(0, 1), a=1, c=Grid(2, 3), d=Grid("a", "b"))
     df2 = ArrayDataFrame([[0, 1], [1, 1], [3, 2]], "a:int,bb:int")
@@ -60,7 +62,7 @@ def assert_count(df: DataFrame, n: int, schema=None) -> None:
     engine = NativeExecutionEngine(conf={TUNE_TEMP_PATH: str(tmpdir)})
 
     # test multiple dfs, batch_size and config
-    with FugueWorkflow(engine) as dag:
+    with FugueWorkflow() as dag:
         dfs = WorkflowDataFrames(
             a=dag.df(df1).partition_by("a"), b=dag.df(df2).partition_by("a")
         )
@@ -90,6 +92,7 @@ def assert_count(df: DataFrame, n: int, schema=None) -> None:
                 f"__tune_df__c:str,{TUNE_DATASET_TRIALS}:str",
             ),
         )
+    dag.run(engine)
 
 
 def test_dataset(tmpdir):

tune_version/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.2"
+__version__ = "0.1.3"