robocandywrapper 0.2.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- robocandywrapper/__init__.py +36 -0
- robocandywrapper/constants.py +14 -0
- robocandywrapper/dataformats/__init__.py +0 -0
- robocandywrapper/dataformats/lerobot_21/__init__.py +32 -0
- robocandywrapper/dataformats/lerobot_21/dataset.py +1077 -0
- robocandywrapper/dataformats/lerobot_21/utils.py +206 -0
- robocandywrapper/factory.py +295 -0
- robocandywrapper/metadata_view.py +280 -0
- robocandywrapper/plugin.py +94 -0
- robocandywrapper/plugins/__init__.py +18 -0
- robocandywrapper/plugins/affordance.py +180 -0
- robocandywrapper/plugins/episode_outcome.py +310 -0
- robocandywrapper/samplers/__init__.py +13 -0
- robocandywrapper/samplers/config.py +120 -0
- robocandywrapper/samplers/factory.py +184 -0
- robocandywrapper/samplers/weighted.py +129 -0
- robocandywrapper/utils.py +127 -0
- robocandywrapper/wrapper.py +475 -0
- robocandywrapper-0.2.1.dist-info/METADATA +122 -0
- robocandywrapper-0.2.1.dist-info/RECORD +23 -0
- robocandywrapper-0.2.1.dist-info/WHEEL +5 -0
- robocandywrapper-0.2.1.dist-info/licenses/LICENSE +21 -0
- robocandywrapper-0.2.1.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
"""RoboCandyWrapper - Extensible dataset wrapper for LeRobot datasets."""
|
|
2
|
+
|
|
3
|
+
from robocandywrapper.plugin import (
|
|
4
|
+
DatasetPlugin,
|
|
5
|
+
PluginInstance,
|
|
6
|
+
PluginConflictError,
|
|
7
|
+
)
|
|
8
|
+
from robocandywrapper.wrapper import WrappedRobotDataset
|
|
9
|
+
from robocandywrapper.metadata_view import WrappedRobotDatasetMetadataView
|
|
10
|
+
from robocandywrapper.samplers.weighted import WeightedSampler
|
|
11
|
+
from robocandywrapper.samplers.factory import make_sampler
|
|
12
|
+
from robocandywrapper.factory import make_dataset_without_config, make_dataset
|
|
13
|
+
from robocandywrapper.utils import WandBLogger
|
|
14
|
+
from robocandywrapper.constants import (
|
|
15
|
+
CANDYWRAPPER_PLUGINS_DIR,
|
|
16
|
+
AFFORDANCE_PLUGIN_NAME,
|
|
17
|
+
EPISODE_OUTCOME_PLUGIN_NAME,
|
|
18
|
+
)
|
|
19
|
+
|
|
20
|
+
__version__ = "0.2.1"
|
|
21
|
+
|
|
22
|
+
__all__ = [
|
|
23
|
+
"DatasetPlugin",
|
|
24
|
+
"PluginInstance",
|
|
25
|
+
"PluginConflictError",
|
|
26
|
+
"WrappedRobotDataset",
|
|
27
|
+
"WrappedRobotDatasetMetadataView",
|
|
28
|
+
"WeightedSampler",
|
|
29
|
+
"make_sampler",
|
|
30
|
+
"make_dataset_without_config",
|
|
31
|
+
"make_dataset",
|
|
32
|
+
"WandBLogger",
|
|
33
|
+
"CANDYWRAPPER_PLUGINS_DIR",
|
|
34
|
+
"AFFORDANCE_PLUGIN_NAME",
|
|
35
|
+
"EPISODE_OUTCOME_PLUGIN_NAME",
|
|
36
|
+
]
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
"""
|
|
2
|
+
Constants for RoboCandyWrapper plugins and data storage.
|
|
3
|
+
"""
|
|
4
|
+
|
|
5
|
+
# Plugin storage directory (relative to dataset root)
|
|
6
|
+
CANDYWRAPPER_PLUGINS_DIR = "candywrapper_plugins"
|
|
7
|
+
|
|
8
|
+
# Plugin-specific subdirectory names
|
|
9
|
+
AFFORDANCE_PLUGIN_NAME = "affordance"
|
|
10
|
+
EPISODE_OUTCOME_PLUGIN_NAME = "episode_outcome"
|
|
11
|
+
# Add more plugin names here as needed
|
|
12
|
+
# GOAL_PLUGIN_NAME = "goal"
|
|
13
|
+
# LANGUAGE_PLUGIN_NAME = "language"
|
|
14
|
+
|
|
File without changes
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
from .dataset import LeRobot21Dataset, LeRobot21DatasetMetadata
|
|
2
|
+
from .utils import (
|
|
3
|
+
load_tasks,
|
|
4
|
+
load_episodes,
|
|
5
|
+
append_jsonlines,
|
|
6
|
+
load_jsonlines,
|
|
7
|
+
serialize_dict,
|
|
8
|
+
cast_stats_to_numpy,
|
|
9
|
+
write_episode,
|
|
10
|
+
write_episode_stats,
|
|
11
|
+
load_episodes_stats,
|
|
12
|
+
backward_compatible_episodes_stats,
|
|
13
|
+
get_episode_data_index,
|
|
14
|
+
check_timestamps_sync,
|
|
15
|
+
)
|
|
16
|
+
|
|
17
|
+
__all__ = [
|
|
18
|
+
"LeRobot21Dataset",
|
|
19
|
+
"LeRobot21DatasetMetadata",
|
|
20
|
+
"load_tasks",
|
|
21
|
+
"load_episodes",
|
|
22
|
+
"append_jsonlines",
|
|
23
|
+
"load_jsonlines",
|
|
24
|
+
"serialize_dict",
|
|
25
|
+
"cast_stats_to_numpy",
|
|
26
|
+
"write_episode",
|
|
27
|
+
"write_episode_stats",
|
|
28
|
+
"load_episodes_stats",
|
|
29
|
+
"backward_compatible_episodes_stats",
|
|
30
|
+
"get_episode_data_index",
|
|
31
|
+
"check_timestamps_sync",
|
|
32
|
+
]
|