Updates

xautodl/datasets/synthetic_env.py

@@ -84,6 +84,14 @@ class SyntheticDEnv(data.Dataset):
     def mode(self):
         return self._time_generator.mode
 
+    def get_seq_times(self, index, seq_length):
+        index, timestamp = self._time_generator[index]
+        xtimes = []
+        for i in range(1, seq_length + 1):
+          xtimes.append(timestamp - i * self.time_interval)
+        xtimes.reverse()
+        return xtimes
+
     def get_timestamp(self, index):
         if index is None:
             timestamps = []