Make sure init_hydra_config does not require any keys (#376)

This commit is contained in:
Alexander Soare
2024-08-23 12:27:08 +01:00
committed by GitHub
parent a2592a5563
commit 9c7649f140
4 changed files with 29 additions and 9 deletions

View File

@@ -1,5 +1,6 @@
import random
from typing import Callable
from uuid import uuid4
import numpy as np
import pytest
@@ -13,6 +14,7 @@ from lerobot.common.datasets.utils import (
)
from lerobot.common.utils.utils import (
get_global_random_state,
init_hydra_config,
seeded_context,
set_global_random_state,
set_global_seed,
@@ -83,3 +85,10 @@ def test_reset_episode_index():
correct_episode_index = [0, 0, 1, 2, 2, 2]
dataset = reset_episode_index(dataset)
assert dataset["episode_index"] == correct_episode_index
def test_init_hydra_config_empty():
test_file = f"/tmp/test_init_hydra_config_empty_{uuid4().hex}.yaml"
with open(test_file, "w") as f:
f.write("\n")
init_hydra_config(test_file)