update test

ahalev · Dec 20, 2023 · f8f80ef · f8f80ef
1 parent 8d7b0a3
commit f8f80ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/envs/test_discrete.py b/tests/envs/test_discrete.py
@@ -11,7 +11,7 @@ class TestDiscreteEnv(TestCase):
 
     def test_init_from_microgrid(self):
         microgrid = get_modular_microgrid()
-        env = DiscreteMicrogridEnv(microgrid)
+        env = DiscreteMicrogridEnv.from_microgrid(microgrid)
 
         self.assertEqual(env.modules, microgrid.modules)
         self.assertIsNot(env.modules.to_tuples(), microgrid.modules.to_tuples())