diff --git a/skais/ais/ais_trajectory.py b/skais/ais/ais_trajectory.py index 895b596cb33fea9fc065ce309c90d8751ef0802e..aa105a756e7e8c5660c02ec95a2c4cc51ee0f715 100644 --- a/skais/ais/ais_trajectory.py +++ b/skais/ais/ais_trajectory.py @@ -5,6 +5,7 @@ import numpy as np from numba import jit from scipy.interpolate import interp1d +from skais.ais.ais_points import AISPoints from skais.process.basic_features_operations import angular_dispersion from skais.utils.geography import great_circle from skais.utils.stats import calc_std_dev @@ -213,7 +214,7 @@ def apply_time_sequence(dat, time, func): return result -class AISTrajectory: +class AISTrajectory(AISPoints): def __init__(self, df, interpolation_time=None): df = df.drop_duplicates(subset=['ts_sec']) @@ -243,7 +244,7 @@ class AISTrajectory: df = new_df # self.df = df.dropna() - self.df = df + AISPoints.__init__(df) def __eq__(self, other): return self.df.equals(other.df)