diff --git a/spatz/sensors/__init__.py b/spatz/sensors/__init__.py index a182a78..8084f96 100644 --- a/spatz/sensors/__init__.py +++ b/spatz/sensors/__init__.py @@ -1,4 +1,8 @@ from spatz.sensors.sensor import Sensor +from spatz.sensors.gps import GPS from spatz.sensors.imu import Accelerometer, Gyroscope, IMU from spatz.sensors.pressure import PressureSensor -from spatz.sensors.compound import CompoundSensor \ No newline at end of file +from spatz.sensors.compound import CompoundSensor + + +from pandas import NA \ No newline at end of file diff --git a/spatz/sensors/gps/__init__.py b/spatz/sensors/gps/__init__.py new file mode 100644 index 0000000..6178f7d --- /dev/null +++ b/spatz/sensors/gps/__init__.py @@ -0,0 +1 @@ +from spatz.sensors.gps.gps import GPS \ No newline at end of file diff --git a/spatz/transforms/transform.py b/spatz/transforms/transform.py index 24d7498..bd67f13 100644 --- a/spatz/transforms/transform.py +++ b/spatz/transforms/transform.py @@ -5,7 +5,7 @@ from typing import Any, Tuple class Transform: - def apply(self, t: float, x: np.array): + def apply(self, t: float, x: ArrayLike): y = self(t, x) assert x.shape == y.shape, "Transform has to maintain the array's shape."