formatting

cpnota · Jan 26, 2024 · 0ccd4c4 · 0ccd4c4
1 parent 0ba62f8
commit 0ccd4c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/all/presets/builder_test.py b/all/presets/builder_test.py
@@ -10,6 +10,7 @@ def setUp(self):
  "lr": 1e-4,
  "gamma": 0.99
  }
+
  class MockPreset():
  def __init__(self, env, name, device, **hyperparameters):
  self.env = env
@@ -48,7 +49,7 @@ def test_override_device(self):
 
  def test_no_side_effects(self):
  self.builder.device('cpu').hyperparameters(lr=0.01).device('cpu').env(Mock).build()
- my_env = Mock 
+ my_env = Mock
  agent = self.builder.env(Mock).build()
  self.assertEqual(agent.name, self.name)
  self.assertEqual(agent.hyperparameters, self.default_hyperparameters)
@@ -61,5 +62,6 @@ def test_call_api(self):
  self.assertEqual(agent.hyperparameters, {**self.default_hyperparameters, "lr": 0.01})
  self.assertEqual(agent.device, 'cpu')
 
+
 if __name__ == "__main__":
  unittest.main()