add top level imports (distributed will import things now) (#331)

src/dask_awkward/__init__.py

@@ -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