XY

2021-05-28 02:40:08 +08:00
parent 89448e433f
commit b8b94cc791
1 changed files with 3 additions and 3 deletions
--- a/tests/test_synthetic_env.py
+++ b/tests/test_synthetic_env.py
@@ -17,4 +17,4 @@ class TestSynethicEnv(unittest.TestCase):
            env = get_synthetic_env(version=version)
            print(env)
            for timestamp, (x, y) in env:
-            self.assertEqual(x.shape, (1000, env.ndim))
+                self.assertEqual(x.shape, (1000, env._data_generator.ndim))