Add more functions for synthetic env

lib/datasets/synthetic_env.py

@@ -55,6 +55,10 @@ class SyntheticDEnv(data.Dataset):
     def timestamp_interval(self):
         return self._timestamp_generator.interval
 
+    def get_timestamp(self, index):
+        index, timestamp = self._timestamp_generator[index]
+        return timestamp
+
     def set_oracle_map(self, functor):
         self._oracle_map = functor
 

lib/datasets/synthetic_utils.py

@@ -60,7 +60,7 @@ class TimeStamp(UnifiedSplit, data.Dataset):
     @property
     def max_timestamp(self):
         return self._max_timestamp
-  
+
     @property
     def interval(self):
         return self._interval