updated notebooks

Author: mxaviersmp

notebooks/01_Exploring_MoveDataFrame.ipynb

@@ -1,6 +1,8 @@
 """MoveDataFrame class."""
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
+
 from dateutil.parser._parser import ParserError
 from pandas.core.frame import DataFrame
 
@@ -13,12 +15,16 @@
     TYPE_PANDAS,
 )
 
+if TYPE_CHECKING:
+    from pymove.core.dask import DaskMoveDataFrame
+    from pymove.core.pandas import PandasMoveDataFrame
+
 
 class MoveDataFrame:
     """Auxiliary class to check and transform data into Pymove Dataframes."""
 
     @staticmethod
-    def __new__(
+    def __new__(  # type: ignore[misc]
         self,
         data: DataFrame | dict | list,
         latitude: str = LATITUDE,
@@ -27,7 +33,7 @@ def __new__(
         traj_id: str = TRAJ_ID,
         type_: str = TYPE_PANDAS,
         n_partitions: int = 1,
-    ):
+    ) -> 'PandasMoveDataFrame' | 'DaskMoveDataFrame':
         """
         Creates the PyMove dataframe, which must contain latitude, longitude and datetime.
 
@@ -68,6 +74,9 @@ def __new__(
             return DaskMoveDataFrame(
                 data, latitude, longitude, datetime, traj_id, n_partitions
             )
+        raise TypeError(
+            f'Unknown MoveDataFrame type {type_}, use {TYPE_PANDAS} or {TYPE_DASK}'
+        )
 
     @staticmethod
     def has_columns(data: DataFrame) -> bool:

notebooks/02_Exploring_Preprossessing.ipynb

@@ -167,7 +167,7 @@ def lng(self) -> Series:
         return self[LONGITUDE]
 
     @property
-    def datetime(self):
+    def datetime(self) -> Series:
         """
         Checks for the DATETIME column and returns its value.