chore: move constants to utils (#2016)
This commit is contained in:
@@ -13,7 +13,6 @@
|
||||
# limitations under the License.
|
||||
from torch.optim.lr_scheduler import LambdaLR
|
||||
|
||||
from lerobot.constants import SCHEDULER_STATE
|
||||
from lerobot.optim.schedulers import (
|
||||
CosineDecayWithWarmupSchedulerConfig,
|
||||
DiffuserSchedulerConfig,
|
||||
@@ -21,6 +20,7 @@ from lerobot.optim.schedulers import (
|
||||
load_scheduler_state,
|
||||
save_scheduler_state,
|
||||
)
|
||||
from lerobot.utils.constants import SCHEDULER_STATE
|
||||
|
||||
|
||||
def test_diffuser_scheduler(optimizer):
|
||||
|
||||
Reference in New Issue
Block a user