Fix bugs

xautodl/datasets/math_base_funcs.py

@@ -28,7 +28,9 @@ class FitFunc(abc.ABC):
         self._params = copy.deepcopy(params)
 
     def check_valid(self):
-        for key, value in self._params.items():
+        # for key, value in self._params.items():
+        for key in range(self._freedom):
+            value = self._params[key]
             if value is None:
                 raise ValueError("The {:} is None".format(key))
 

xautodl/datasets/synthetic_core.py

@@ -44,6 +44,7 @@ def get_synthetic_env(total_timestamp=1600, num_per_task=1000, mode=None, versio
             params={
                 0: LinearFunc(params={0: 0.1, 1: 0}),  # 0.1 * t
                 1: SinFunc(params={0: 1, 1: 1, 2: 0}),  # sin(t)
+                2: ConstantFunc(0),
             }
         )
         dynamic_env = SyntheticDEnv(