SPATZ migration + proper directory structure

This commit is contained in:
dario
2023-12-10 14:11:54 +01:00
parent f819b24bfa
commit c60629b4c9
24 changed files with 938 additions and 0 deletions

22
spatz/sensors/compound.py Normal file
View File

@@ -0,0 +1,22 @@
import numpy as np
from typing import List
from numpy.typing import ArrayLike
from sensor import Sensor
from spatz.dataset import Dataset, List
from spatz.logger import Logger
from spatz.transforms import Transform
class CompoundSensor(Sensor):
def __init__(self, dataset: Dataset, logger: Logger, sensors: List[Sensor], transforms: List[Transform] = []):
super().__init__(dataset, logger, transforms)
self.__sensors = sensors
def _get_data(self) -> ArrayLike:
x = np.stack([sensor() for sensor in self.__sensors])
x = self._sensor_specific_effects(x)
return x