Major rework using rocketpy

spatz/sensors/sensor.py

@@ -9,17 +9,24 @@ from pandas import NA
 from spatz.transforms import *
 from spatz.logger import *
 from spatz.dataset import *
+from spatz.simulations.data_source import DataSource
 
 
 class Sensor:
-    def __init__(self, dataset: Dataset, logger: Logger, transforms: List[Transform] = [], min_value=-np.inf, max_value=np.inf):
+    def __init__(
+            self, 
+            dataset: DataSource, 
+            logger: Logger, 
+            transforms: List[Transform] = [], 
+            min_value=-np.inf, 
+            max_value=np.inf):
         self._dataset = dataset
         self._logger = logger
         self._transforms = transforms
         self._min_value = min_value
         self._max_value = max_value
 
-    def set_dataset(self, dataset: Dataset):
+    def set_dataset(self, dataset: DataSource):
         self._dataset = dataset
 
     def set_logger(self, logger: Logger):