Fix bugs in new env-v1

xautodl/datasets/math_dynamic_generator.py

@@ -21,6 +21,8 @@ class DynamicGenerator(abc.ABC):
 
 
 class GaussianDGenerator(DynamicGenerator):
+    """Generate data from Gaussian distribution."""
+
     def __init__(self, mean_functors, cov_functors, trunc=(-1, 1)):
         super(GaussianDGenerator, self).__init__()
         self._ndim = assert_list_tuple(mean_functors)
@@ -41,6 +43,10 @@ class GaussianDGenerator(DynamicGenerator):
             assert assert_list_tuple(trunc) == 2 and trunc[0] < trunc[1]
         self._trunc = trunc
 
+    @property
+    def ndim(self):
+        return self._ndim
+
     def __call__(self, time, num):
         mean_list = [functor(time) for functor in self._mean_functors]
         cov_matrix = [

xautodl/datasets/synthetic_env.py

@@ -115,7 +115,7 @@ class SyntheticDEnv(data.Dataset):
             name=self.__class__.__name__,
             cur_num=len(self),
             total=len(self._time_generator),
-            ndim=self._ndim,
+            ndim=self._data_generator.ndim,
             num_per_task=self._num_per_task,
             xrange_min=self.min_timestamp,
             xrange_max=self.max_timestamp,