fix formatting

lincc-frameworks · Mar 20, 2024 · 35108c3 · 35108c3
1 parent 10be686
commit 35108c3
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 20 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,8 +18,8 @@ requires-python = ">=3.9"
 dependencies = [
     'pandas',
     'numpy',
-    'dask',
-    'dask[distributed]',
+    'dask>=2024.3.0',
+    'dask[distributed]>=2024.3.0',
     'pyarrow',
     'pyvo',
     'scipy',

diff --git a/src/tape/__init__.py b/src/tape/__init__.py
@@ -1,6 +1,5 @@
 from .analysis import *  # noqa
 from .ensemble import *  # noqa
-#from .ensemble_frame import *  # noqa
 from .expr import *  # noqa
 from .timeseries import *  # noqa
 from .ensemble_readers import *  # noqa

diff --git a/tests/tape_tests/conftest.py b/tests/tape_tests/conftest.py
@@ -304,31 +304,23 @@ def parquet_ensemble_with_divisions():
     """Create an Ensemble from parquet data."""
     ens = Ensemble(client=False)
 
-    source_ddf = dd.read_parquet("tests/tape_tests/data/source/test_source.parquet",
-                                 calculate_divisions=True).repartition(npartitions=3)
-    object_ddf = dd.read_parquet("tests/tape_tests/data/object/test_object.parquet",
-                                 calculate_divisions=True).repartition(npartitions=2)
+    source_ddf = dd.read_parquet(
+        "tests/tape_tests/data/source/test_source.parquet", calculate_divisions=True
+    ).repartition(npartitions=3)
+    object_ddf = dd.read_parquet(
+        "tests/tape_tests/data/object/test_object.parquet", calculate_divisions=True
+    ).repartition(npartitions=2)
 
-    ens.from_dask_dataframe(source_ddf,
+    ens.from_dask_dataframe(
+        source_ddf,
         object_ddf,
         id_col="ps1_objid",
         time_col="midPointTai",
         band_col="filterName",
         flux_col="psFlux",
         err_col="psFluxErr",
-        sorted=True,)
-    """
-    ens.from_parquet(
-        "tests/tape_tests/data/source/test_source.parquet",
-        "tests/tape_tests/data/object/test_object.parquet",
-        id_col="ps1_objid",
-        time_col="midPointTai",
-        band_col="filterName",
-        flux_col="psFlux",
-        err_col="psFluxErr",
-        sort=True,
+        sorted=True,
     )
-    """
 
     return ens