dataaug-platform/dataaug_platform/__init__.py at 701013e236db1073ebb3fc01ca7a6c3407315d80 · pairlab/dataaug-platform · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
"""
DataAug Platform - A PySpark-based data augmentation framework for trajectory data.
"""

from .augmentations.base_augmentation import (
    Augmentation,
    local_aug,
    global_aug,
)
from .dataset import SparkIterableDataset
from .ingestion import load_hdf5_group, hdf5_to_rdd, read_hdf5_metadata, write_trajectories_to_hdf5
from .pipeline import Pipeline

__all__ = [
    # augmentation
    "Augmentation",
    "local_aug",
    "global_aug",
    # dataset
    "SparkIterableDataset",
    # ingestion
    "load_hdf5_group",
    "hdf5_to_rdd",
    "read_hdf5_metadata",
    "write_trajectories_to_hdf5",
    # Pipeline
    "Pipeline",
]

# Optional imports - only available if mimicgen dependencies are installed
try:
    from .augmentations.mimicgen import MimicGenAugmentation
    __all__.append("MimicGenAugmentation")
except ImportError:
    pass

__version__ = "0.1.0"