-
Notifications
You must be signed in to change notification settings - Fork 19
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add top level imports (distributed will import things now) (#331)
- Loading branch information
1 parent
75fee25
commit 0c3493d
Showing
1 changed file
with
86 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,19 +1,89 @@ | ||
from dask_awkward import config # isort:skip; load awkward config | ||
|
||
|
||
from dask_awkward.lib.core import Array, PartitionCompatibility, Record, Scalar | ||
from dask_awkward.lib.core import _type as type | ||
from dask_awkward.lib.core import ( | ||
compatible_partitions, | ||
map_partitions, | ||
partition_compatibility, | ||
typetracer_from_form, | ||
) | ||
from dask_awkward.lib.describe import fields | ||
from dask_awkward.lib.inspect import necessary_columns, sample | ||
from dask_awkward.lib.io.io import ( | ||
ImplementsFormTransformation, | ||
from_awkward, | ||
from_dask_array, | ||
from_delayed, | ||
from_lists, | ||
from_map, | ||
to_dask_array, | ||
to_dask_bag, | ||
to_dataframe, | ||
to_delayed, | ||
) | ||
from dask_awkward.lib.io.json import from_json, to_json | ||
from dask_awkward.lib.io.parquet import from_parquet, to_parquet | ||
from dask_awkward.lib.operations import concatenate | ||
from dask_awkward.lib.reducers import ( | ||
all, | ||
any, | ||
argmax, | ||
argmin, | ||
corr, | ||
count, | ||
count_nonzero, | ||
covar, | ||
linear_fit, | ||
max, | ||
mean, | ||
min, | ||
moment, | ||
prod, | ||
ptp, | ||
softmax, | ||
std, | ||
sum, | ||
var, | ||
) | ||
from dask_awkward.lib.structure import ( | ||
argcartesian, | ||
argcombinations, | ||
argsort, | ||
broadcast_arrays, | ||
cartesian, | ||
combinations, | ||
copy, | ||
drop_none, | ||
fill_none, | ||
firsts, | ||
flatten, | ||
from_regular, | ||
full_like, | ||
is_none, | ||
isclose, | ||
local_index, | ||
mask, | ||
nan_to_num, | ||
num, | ||
ones_like, | ||
pad_none, | ||
ravel, | ||
run_lengths, | ||
singletons, | ||
sort, | ||
strings_astype, | ||
to_packed, | ||
to_regular, | ||
unflatten, | ||
unzip, | ||
values_astype, | ||
where, | ||
with_field, | ||
with_name, | ||
with_parameter, | ||
without_parameters, | ||
zeros_like, | ||
zip, | ||
) | ||
from dask_awkward.version import __version__ | ||
|
||
|
||
def __getattr__(value): | ||
import dask_awkward.lib | ||
|
||
return getattr(dask_awkward.lib, value) | ||
|
||
|
||
original = dir() | ||
|
||
|
||
def __dir__(): | ||
import dask_awkward.lib # pragma: no cover | ||
|
||
return original + dir(dask_awkward.lib) # pragma: no cover |