Skip to content

Commit

Permalink
formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
alex-rakowski committed Jun 25, 2024
1 parent 4cfd6cf commit cb50d0c
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions dask_snowflake/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,13 @@

import dask
import dask.dataframe as dd
from dask.base import tokenize
from dask.dataframe.core import new_dd_object

# from dask.base import tokenize
# from dask.dataframe.core import new_dd_object
from dask.delayed import delayed
from dask.highlevelgraph import HighLevelGraph
from dask.layers import DataFrameIOLayer

# from dask.highlevelgraph import HighLevelGraph
# from dask.layers import DataFrameIOLayer
from dask.utils import parse_bytes


Expand Down Expand Up @@ -317,11 +319,10 @@ def read_snowflake(
return dd.from_map(
partial(_fetch_batches, arrow_options=arrow_options),
batches_partitioned,
npartitions = divisions,
npartitions=divisions,
meta=meta,
)


# # Create Blockwise layer
# layer = DataFrameIOLayer(
# output_name,
Expand Down

0 comments on commit cb50d0c

Please sign in to comment.