From 811abce6cf6c8952a6e1328bb2f0eb3151e1ae3c Mon Sep 17 00:00:00 2001 From: David Hoeller Date: Sun, 2 Jun 2024 11:01:29 +0200 Subject: [PATCH] Update unit tests --- .../test/wrappers/test_rl_games_wrapper.py | 4 +--- .../omni.isaac.lab_tasks/test/wrappers/test_rsl_rl_wrapper.py | 4 +--- .../omni.isaac.lab_tasks/test/wrappers/test_sb3_wrapper.py | 4 +--- .../omni.isaac.lab_tasks/test/wrappers/test_skrl_wrapper.py | 4 +--- 4 files changed, 4 insertions(+), 12 deletions(-) diff --git a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rl_games_wrapper.py b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rl_games_wrapper.py index 1609ec982c..0da9baad9a 100644 --- a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rl_games_wrapper.py +++ b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rl_games_wrapper.py @@ -15,7 +15,6 @@ """Rest everything follows.""" import gymnasium as gym -import random import torch import unittest @@ -42,8 +41,7 @@ def setUpClass(cls): cls.registered_tasks.append(task_spec.id) # sort environments by name cls.registered_tasks.sort() - # pick four environments randomly to test - cls.registered_tasks = random.sample(cls.registered_tasks, 4) + cls.registered_tasks = cls.registered_tasks[:4] # print all existing task names print(">>> All registered environments:", cls.registered_tasks) diff --git a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rsl_rl_wrapper.py b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rsl_rl_wrapper.py index eb90e09c43..290f6b944c 100644 --- a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rsl_rl_wrapper.py +++ b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_rsl_rl_wrapper.py @@ -15,7 +15,6 @@ """Rest everything follows.""" import gymnasium as gym -import random import torch import unittest @@ -42,8 +41,7 @@ def setUpClass(cls): cls.registered_tasks.append(task_spec.id) # sort environments by name cls.registered_tasks.sort() - # pick four environments randomly to test - cls.registered_tasks = random.sample(cls.registered_tasks, 4) + cls.registered_tasks = cls.registered_tasks[:4] # print all existing task names print(">>> All registered environments:", cls.registered_tasks) diff --git a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_sb3_wrapper.py b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_sb3_wrapper.py index 3114d346cf..4f1700d7a0 100644 --- a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_sb3_wrapper.py +++ b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_sb3_wrapper.py @@ -16,7 +16,6 @@ import gymnasium as gym import numpy as np -import random import torch import unittest @@ -43,8 +42,7 @@ def setUpClass(cls): cls.registered_tasks.append(task_spec.id) # sort environments by name cls.registered_tasks.sort() - # pick four environments randomly to test - cls.registered_tasks = random.sample(cls.registered_tasks, 4) + cls.registered_tasks = cls.registered_tasks[:4] # print all existing task names print(">>> All registered environments:", cls.registered_tasks) diff --git a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_skrl_wrapper.py b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_skrl_wrapper.py index d6f4a164b8..12b5ef782d 100644 --- a/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_skrl_wrapper.py +++ b/source/extensions/omni.isaac.lab_tasks/test/wrappers/test_skrl_wrapper.py @@ -15,7 +15,6 @@ """Rest everything follows.""" import gymnasium as gym -import random import torch import unittest @@ -42,8 +41,7 @@ def setUpClass(cls): cls.registered_tasks.append(task_spec.id) # sort environments by name cls.registered_tasks.sort() - # pick four environments randomly to test - cls.registered_tasks = random.sample(cls.registered_tasks, 4) + cls.registered_tasks = cls.registered_tasks[:4] # print all existing task names print(">>> All registered environments:", cls.registered_tasks)