wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add todo in skipped test

This commit is contained in:
Simon Alibert 2024-11-02 20:01:34 +01:00
parent 7a342db9c4
commit df2cb51364
1 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/test_delta_timestamps.py
View File

@ -162,20 +162,24 @@ def test_check_timestamps_sync_single_timestamp():
assert result is True assert result is True
# TODO(aliberts): change behavior of hf_transform_to_torch so that it can work with empty dataset # TODO(aliberts): Change behavior of hf_transform_to_torch so that it can work with empty dataset
# def test_check_timestamps_sync_empty_dataset(): @pytest.mark.skip("TODO: fix")
# fps = 30 def test_check_timestamps_sync_empty_dataset():
# tolerance_s = 1e-4 fps = 30
# empty_hf_dataset = Dataset.from_dict({'timestamp': [], 'episode_index': []}) tolerance_s = 1e-4
# empty_hf_dataset.set_transform(hf_transform_to_torch) empty_hf_dataset = Dataset.from_dict({"timestamp": [], "episode_index": []})
# episode_data_index = {'to': torch.tensor([], dtype=torch.int64), 'from': torch.tensor([], dtype=torch.int64)} empty_hf_dataset.set_transform(hf_transform_to_torch)
# result = check_timestamps_sync( episode_data_index = {
# hf_dataset=empty_hf_dataset, "to": torch.tensor([], dtype=torch.int64),
# episode_data_index=episode_data_index, "from": torch.tensor([], dtype=torch.int64),
# fps=fps, }
# tolerance_s=tolerance_s, result = check_timestamps_sync(
# ) hf_dataset=empty_hf_dataset,
# assert result is True episode_data_index=episode_data_index,
fps=fps,
tolerance_s=tolerance_s,
)
assert result is True
def test_check_delta_timestamps_valid(valid_delta_timestamps_factory): def test_check_delta_timestamps_valid(valid_delta_timestamps_factory):