fix pre-commit

automl · Dec 11, 2023 · 897f373 · 897f373
1 parent 4eca801
commit 897f373
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 39 deletions.
diff --git a/carl/envs/gymnasium/carl_gymnasium_env.py b/carl/envs/gymnasium/carl_gymnasium_env.py
@@ -10,10 +10,10 @@
 
 try:
     pygame.display.init()
-except: # pragma: no cover
-    import os # pragma: no cover
+except:  # pragma: no cover
+    import os  # pragma: no cover
 
-    os.environ["SDL_VIDEODRIVER"] = "dummy" # pragma: no cover
+    os.environ["SDL_VIDEODRIVER"] = "dummy"  # pragma: no cover
 
 
 class CARLGymnasiumEnv(CARLEnv):

diff --git a/test/test_context_sampler.py b/test/test_context_sampler.py
@@ -1,5 +1,4 @@
 import unittest
-from omegaconf import DictConfig 
 
 from carl.context.context_space import (
     ContextSpace,
@@ -47,7 +46,7 @@ def test_init(self):
 
         with self.assertRaises(ValueError):
             ContextSampler(
-                context_distributions=0, 
+                context_distributions=0,
                 context_space=self.cspace,
                 seed=0,
                 name="TestSampler",
@@ -63,7 +62,5 @@ def test_sample_contexts(self):
         self.assertEqual(contexts[0]["gravity"], 9.8)
 
 
-
-
 if __name__ == "__main__":
     unittest.main()
diff --git a/test/test_context_space.py b/test/test_context_space.py
@@ -108,5 +108,6 @@ def test_sample(self):
         with self.assertRaises(ValueError):
             self.context_space.sample_contexts(["false_feature"], size=0)
 
+
 if __name__ == "__main__":
     unittest.main()
diff --git a/test/test_search_space_encoding.py b/test/test_search_space_encoding.py
@@ -5,31 +5,37 @@
 
 from carl.context.search_space_encoding import search_space_to_config_space
 
-
 dict_space = {
-                "uniform_integer": (1, 10),
-                "uniform_float": (1.0, 10.0),
-                "categorical": ["a", "b", "c"],
-                "constant": 1337,
-            }
+    "uniform_integer": (1, 10),
+    "uniform_float": (1.0, 10.0),
+    "categorical": ["a", "b", "c"],
+    "constant": 1337,
+}
 
-dict_space_2 = { "hyperparameters": [
-    {"name": "x0", 
-    "type": "uniform_float", 
-    "log": False,
-    "lower": -512.0,
-    "upper": 512.0, 
-    "default": -3.0},
-    {"name": "x1", 
-    "type": "uniform_float",
-    "log": False,
-    "lower": -512.0,
-    "upper": 512.0,
-    "default": -4.0}],
+dict_space_2 = {
+    "hyperparameters": [
+        {
+            "name": "x0",
+            "type": "uniform_float",
+            "log": False,
+            "lower": -512.0,
+            "upper": 512.0,
+            "default": -3.0,
+        },
+        {
+            "name": "x1",
+            "type": "uniform_float",
+            "log": False,
+            "lower": -512.0,
+            "upper": 512.0,
+            "default": -4.0,
+        },
+    ],
     "conditions": [],
     "forbiddens": [],
     "python_module_version": "0.4.17",
-    "json_format_version": 0.2}
+    "json_format_version": 0.2,
+}
 
 str_space = """{
                 "uniform_integer": (1, 10),
@@ -38,25 +44,17 @@
                 "constant": 1337,
             }"""
 
+
 class TestSearchSpacEncoding(unittest.TestCase):
     def setUp(self):
         self.test_space = None
-        self.test_space = ConfigurationSpace(
-            name="myspace",
-            space=dict_space
-        )
+        self.test_space = ConfigurationSpace(name="myspace", space=dict_space)
         return super().setUp()
 
     def test_init(self):
-        self.test_space = ConfigurationSpace(
-            name="myspace",
-            space=dict_space_2
-        )
+        self.test_space = ConfigurationSpace(name="myspace", space=dict_space_2)
 
-        self.test_space = ConfigurationSpace(
-            name="myspace",
-            space=str_space
-        )
+        self.test_space = ConfigurationSpace(name="myspace", space=str_space)
 
     def test_config_spaces(self):
         try: