Update unit tests

isaac-sim · Jun 2, 2024 · 811abce · 811abce
1 parent 800042b
commit 811abce
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 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
@@ -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
@@ -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
@@ -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
@@ -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)