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
@@ -15,6 +15,6 @@ class TestSynethicEnv(unittest.TestCase):
        versions = ["v1", "v2", "v3", "v4"]
        for version in versions:
            env = get_synthetic_env(version=version)
-        print(env)
-        for timestamp, (x, y) in env:
-            self.assertEqual(x.shape, (1000, env.ndim))
+            print(env)
+            for timestamp, (x, y) in env:
+                self.assertEqual(x.shape, (1000, env._data_generator.ndim))