meerschaum 2.6.0__tar.gz → 2.6.1__tar.gz

{meerschaum-2.6.0/meerschaum.egg-info → meerschaum-2.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: meerschaum
-Version: 2.6.0
+Version: 2.6.1
 Summary: Sync Time-Series Pipes with Meerschaum
 Home-page: https://meerschaum.io
 Author: Bennett Meares

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/api/routes/_pipes.py

@@ -7,8 +7,16 @@ Register Pipes via the Meerschaum API.
 """
 
 from __future__ import annotations
-from meerschaum.utils.typing import Any, Optional, Dict, Union
 
+
+import io
+import json
+import fastapi
+from decimal import Decimal
+import datetime
+
+import meerschaum as mrsm
+from meerschaum.utils.typing import Any, Optional, Dict, Union, List
 from meerschaum.api import (
     fastapi,
     app,
@@ -19,27 +27,27 @@ from meerschaum.api import (
     _get_pipes,
     manager,
     debug,
-    no_auth, private,
+    no_auth,
+    private,
 )
-import json
-import fastapi
-from decimal import Decimal
 from meerschaum import Pipe
 from meerschaum.api.models import MetaPipe
 from meerschaum.utils.packages import attempt_import, import_pandas
-from meerschaum.utils.dataframe import get_numeric_cols, to_json
+from meerschaum.utils.dataframe import get_numeric_cols, to_json, parse_df_datetimes
+from meerschaum.utils.dtypes import are_dtypes_equal
 from meerschaum.utils.misc import (
-    is_pipe_registered, round_time, is_int, parse_df_datetimes,
+    is_pipe_registered,
+    round_time,
+    is_int,
     replace_pipes_in_dict,
 )
-from meerschaum.utils.typing import List, Dict, Any, Union
 import meerschaum.core.User
-import datetime
-pipes_endpoint = endpoints['pipes']
-from fastapi.responses import StreamingResponse
-import io
+
+fastapi_responses = attempt_import('fastapi.responses', lazy=False)
+StreamingResponse = fastapi_responses.StreamingResponse
 dateutil_parser = attempt_import('dateutil.parser', lazy=False)
-pd = attempt_import('pandas')
+pipes_endpoint = endpoints['pipes']
+pd = attempt_import('pandas', lazy=False)
 
 
 @app.post(pipes_endpoint + '/{connector_keys}/{metric_key}/{location_key}/register', tags=['Pipes'])
@@ -79,13 +87,13 @@ def register_pipe(
 
 @app.delete(pipes_endpoint + '/{connector_keys}/{metric_key}/{location_key}/delete', tags=['Pipes'])
 def delete_pipe(
-        connector_keys: str,
-        metric_key: str,
-        location_key: str,
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-    ):
+    connector_keys: str,
+    metric_key: str,
+    location_key: str,
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+):
     """
     Delete a Pipe (without dropping its table).
     """
@@ -141,15 +149,15 @@ def drop_pipe(
 
 @app.patch(pipes_endpoint + '/{connector_keys}/{metric_key}/{location_key}/edit', tags=['Pipes'])
 def edit_pipe(
-        connector_keys: str,
-        metric_key: str,
-        location_key: str,
-        parameters: dict,
-        patch: bool = False,
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-    ):
+    connector_keys: str,
+    metric_key: str,
+    location_key: str,
+    parameters: dict,
+    patch: bool = False,
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+):
     """
     Edit an existing pipe.
     """
@@ -177,15 +185,15 @@ def edit_pipe(
 
 @app.get(pipes_endpoint + '/keys', tags=['Pipes'])
 async def fetch_pipes_keys(
-        connector_keys: str = "[]",
-        metric_keys: str = "[]",
-        location_keys: str = "[]",
-        tags: str = "[]",
-        params: str = "{}",
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-    ):
+    connector_keys: str = "[]",
+    metric_keys: str = "[]",
+    location_keys: str = "[]",
+    tags: str = "[]",
+    params: str = "{}",
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+):
     """
     Get a list of tuples of all registered Pipes' keys.
     """
@@ -201,14 +209,14 @@ async def fetch_pipes_keys(
 
 @app.get(pipes_endpoint, tags=['Pipes'])
 async def get_pipes(
-        connector_keys: str = "",
-        metric_keys: str = "",
-        location_keys: str = "",
-        curr_user=(
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-        debug: bool = False,
-    ) -> Dict[str, Any]:
+    connector_keys: str = "",
+    metric_keys: str = "",
+    location_keys: str = "",
+    curr_user=(
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+    debug: bool = False,
+) -> Dict[str, Any]:
     """
     Get all registered Pipes with metadata, excluding parameters.
     """
@@ -223,11 +231,11 @@ async def get_pipes(
 
 @app.get(pipes_endpoint + '/{connector_keys}', tags=['Pipes'])
 async def get_pipes_by_connector(
-        connector_keys: str,
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-    ) -> Dict[str, Any]:
+    connector_keys: str,
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+) -> Dict[str, Any]:
     """
     Get all registered Pipes by connector_keys with metadata, excluding parameters.
     """
@@ -240,13 +248,13 @@ async def get_pipes_by_connector(
 
 @app.get(pipes_endpoint + '/{connector_keys}/{metric_key}', tags=['Pipes'])
 async def get_pipes_by_connector_and_metric(
-        connector_keys: str,
-        metric_key: str,
-        parent: bool = False,
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-    ):
+    connector_keys: str,
+    metric_key: str,
+    parent: bool = False,
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+):
     """
     Get all registered Pipes by connector_keys and metric_key with metadata, excluding parameters.
 
@@ -269,13 +277,13 @@ async def get_pipes_by_connector_and_metric(
 
 @app.get(pipes_endpoint + '/{connector_keys}/{metric_key}/{location_key}', tags=['Pipes'])
 async def get_pipes_by_connector_and_metric_and_location(
-        connector_keys: str,
-        metric_key: str,
-        location_key: str,
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-    ):
+    connector_keys: str,
+    metric_key: str,
+    location_key: str,
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+):
     """
     Get a specific Pipe with metadata, excluding parameters.
     """
@@ -297,17 +305,17 @@ async def get_pipes_by_connector_and_metric_and_location(
 
 @app.get(pipes_endpoint + '/{connector_keys}/{metric_key}/{location_key}/sync_time', tags=['Pipes'])
 def get_sync_time(
-        connector_keys: str,
-        metric_key: str,
-        location_key: str,
-        params: Optional[Dict[str, Any]] = None,
-        newest: bool = True,
-        round_down: bool = True,
-        debug: bool = False,
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-    ) -> Union[str, int, None]:
+    connector_keys: str,
+    metric_key: str,
+    location_key: str,
+    params: Optional[Dict[str, Any]] = None,
+    newest: bool = True,
+    round_down: bool = True,
+    debug: bool = False,
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+) -> Union[str, int, None]:
     """
     Get a Pipe's latest datetime value.
     See `meerschaum.Pipe.get_sync_time`.
@@ -328,20 +336,20 @@ def get_sync_time(
 
 @app.post(pipes_endpoint + '/{connector_keys}/{metric_key}/{location_key}/data', tags=['Pipes'])
 def sync_pipe(
-        connector_keys: str,
-        metric_key: str,
-        location_key: str,
-        data: dict = None,
-        check_existing: bool = True,
-        blocking: bool = True,
-        force: bool = False,
-        workers: Optional[int] = None,
-        columns: Optional[str] = None,
-        curr_user = (
-            fastapi.Depends(manager) if not no_auth else None
-        ),
-        debug: bool = False,
-    ) -> List[Union[bool, str]]:
+    connector_keys: str,
+    metric_key: str,
+    location_key: str,
+    data: dict = None,
+    check_existing: bool = True,
+    blocking: bool = True,
+    force: bool = False,
+    workers: Optional[int] = None,
+    columns: Optional[str] = None,
+    curr_user = (
+        fastapi.Depends(manager) if not no_auth else None
+    ),
+    debug: bool = False,
+) -> List[Union[bool, str]]:
     """
     Add data to an existing Pipe.
     See `meerschaum.Pipe.sync`.
@@ -365,11 +373,11 @@ def sync_pipe(
 
     result = list(p.sync(
         data,
-        debug = debug,
-        check_existing = check_existing,
-        blocking = blocking,
-        force = force,
-        workers = workers
+        debug=debug,
+        check_existing=check_existing,
+        blocking=blocking,
+        force=force,
+        workers=workers,
     ))
     return result
 
@@ -581,7 +589,7 @@ def get_pipe_attributes(
     ),
 ) -> Dict[str, Any]:
     """Get a Pipe's attributes."""
-    return get_pipe(connector_keys, metric_key, location_key).attributes
+    return get_pipe(connector_keys, metric_key, location_key, refresh=True).attributes
 
 
 @app.get(pipes_endpoint + '/{connector_keys}/{metric_key}/{location_key}/exists', tags=['Pipes'])

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/config/_version.py

@@ -2,4 +2,4 @@
 Specify the Meerschaum release version.
 """
 
-__version__ = "2.6.0"
+__version__ = "2.6.1"

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/connectors/api/_pipes.py

@@ -367,6 +367,7 @@ def get_pipe_data(
 
     from meerschaum.utils.packages import import_pandas
     from meerschaum.utils.dataframe import parse_df_datetimes, add_missing_cols_to_df
+    from meerschaum.utils.dtypes import are_dtypes_equal
     pd = import_pandas()
     try:
         df = pd.read_json(StringIO(response.text))
@@ -382,16 +383,17 @@ def get_pipe_data(
         ignore_cols = [
             col
             for col, dtype in pipe.dtypes.items()
-            if 'datetime' not in str(dtype)
+            if not are_dtypes_equal(str(dtype), 'datetime')
         ],
-        debug = debug,
+        strip_timezone=(pipe.tzinfo is None),
+        debug=debug,
     )
     return df
 
 
 def get_pipe_id(
     self,
-    pipe: meerschuam.Pipe,
+    pipe: mrsm.Pipe,
     debug: bool = False,
 ) -> int:
     """Get a Pipe's ID from the API."""
@@ -413,7 +415,7 @@ def get_pipe_id(
 
 def get_pipe_attributes(
     self,
-    pipe: meerschaum.Pipe,
+    pipe: mrsm.Pipe,
     debug: bool = False,
 ) -> Dict[str, Any]:
     """Get a Pipe's attributes from the API
@@ -439,7 +441,7 @@ def get_pipe_attributes(
 
 def get_sync_time(
     self,
-    pipe: 'meerschaum.Pipe',
+    pipe: mrsm.Pipe,
     params: Optional[Dict[str, Any]] = None,
     newest: bool = True,
     debug: bool = False,

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/connectors/sql/_fetch.py

@@ -90,16 +90,18 @@ def fetch(
     )
     ### if sqlite, parse for datetimes
     if not as_hook_results and self.flavor == 'sqlite':
-        from meerschaum.utils.misc import parse_df_datetimes
+        from meerschaum.utils.dataframe import parse_df_datetimes
+        from meerschaum.utils.dtypes import are_dtypes_equal
         ignore_cols = [
             col
             for col, dtype in pipe.dtypes.items()
-            if 'datetime' not in str(dtype)
+            if not are_dtypes_equal(str(dtype), 'datetime')
         ]
         return (
             parse_df_datetimes(
                 chunk,
                 ignore_cols=ignore_cols,
+                strip_timezone=(pipe.tzinfo is None),
                 debug=debug,
             )
             for chunk in chunks

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/connectors/sql/_pipes.py

@@ -872,7 +872,11 @@ def get_pipe_data(
     from meerschaum.utils.sql import sql_item_name
     from meerschaum.utils.misc import parse_df_datetimes, to_pandas_dtype
     from meerschaum.utils.packages import import_pandas
-    from meerschaum.utils.dtypes import attempt_cast_to_numeric, attempt_cast_to_uuid
+    from meerschaum.utils.dtypes import (
+        attempt_cast_to_numeric,
+        attempt_cast_to_uuid,
+        are_dtypes_equal,
+    )
     pd = import_pandas()
     is_dask = 'dask' in pd.__name__
 
@@ -967,7 +971,7 @@ def get_pipe_data(
         ignore_dt_cols = [
             col
             for col, dtype in pipe.dtypes.items()
-            if 'datetime' not in str(dtype)
+            if not are_dtypes_equal(str(dtype), 'datetime')
         ]
         ### NOTE: We have to consume the iterator here to ensure that datetimes are parsed correctly
         df = (
@@ -975,6 +979,7 @@ def get_pipe_data(
                 df,
                 ignore_cols=ignore_dt_cols,
                 chunksize=kw.get('chunksize', None),
+                strip_timezone=(pipe.tzinfo is None),
                 debug=debug,
             ) if isinstance(df, pd.DataFrame) else (
                 [
@@ -982,6 +987,7 @@ def get_pipe_data(
                         c,
                         ignore_cols=ignore_dt_cols,
                         chunksize=kw.get('chunksize', None),
+                        strip_timezone=(pipe.tzinfo is None),
                         debug=debug,
                     )
                     for c in df
@@ -1090,6 +1096,8 @@ def get_pipe_data_query(
         if begin is not None:
             begin -= backtrack_interval
 
+    begin, end = pipe.parse_date_bounds(begin, end)
+
     cols_names = [
         sql_item_name(col, self.flavor, None)
         for col in select_columns
@@ -1661,11 +1669,14 @@ def sync_pipe(
                 'autoincrement': False,
             },
         )
-        temp_pipe._columns_types = {
-            col: get_db_type_from_pd_type(str(typ), self.flavor)
+        temp_pipe.__dict__['_columns_types'] = {
+            col: get_db_type_from_pd_type(
+                pipe.dtypes.get(col, str(typ)),
+                self.flavor,
+            )
             for col, typ in update_df.dtypes.items()
         }
-        temp_pipe._columns_types_timestamp = time.perf_counter()
+        temp_pipe.__dict__['_columns_types_timestamp'] = time.perf_counter()
         temp_success, temp_msg = temp_pipe.sync(update_df, check_existing=False, debug=debug)
         if not temp_success:
             return temp_success, temp_msg

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/connectors/valkey/_pipes.py

@@ -409,6 +409,7 @@ def get_pipe_data(
         return None
 
     from meerschaum.utils.dataframe import query_df, parse_df_datetimes
+    from meerschaum.utils.dtypes import are_dtypes_equal
 
     valkey_dtypes = pipe.parameters.get('valkey', {}).get('dtypes', {})
     dt_col = pipe.columns.get('datetime', None)
@@ -442,13 +443,14 @@ def get_pipe_data(
     ignore_dt_cols = [
         col
         for col, dtype in pipe.dtypes.items()
-        if 'datetime' not in str(dtype)
+        if not are_dtypes_equal(str(dtype), 'datetime')
     ]
 
     df = parse_df_datetimes(
         docs,
         ignore_cols=ignore_dt_cols,
         chunksize=kwargs.get('chunksize', None),
+        strip_timezone=(pipe.tzinfo is None),
         debug=debug,
     )
     for col, typ in valkey_dtypes.items():

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/core/Pipe/__init__.py

@@ -105,6 +105,7 @@ class Pipe:
         autoincrement,
         upsert,
         static,
+        tzinfo,
         get_columns,
         get_columns_types,
         get_columns_indices,

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/core/Pipe/_attributes.py

@@ -8,10 +8,13 @@ Fetch and manipulate Pipes' attributes
 
 from __future__ import annotations
 
+from datetime import timezone
+
 import meerschaum as mrsm
 from meerschaum.utils.typing import Tuple, Dict, SuccessTuple, Any, Union, Optional, List
 from meerschaum.utils.warnings import warn
 
+
 @property
 def attributes(self) -> Dict[str, Any]:
     """
@@ -52,6 +55,13 @@ def parameters(self) -> Optional[Dict[str, Any]]:
     """
     if 'parameters' not in self.attributes:
         self.attributes['parameters'] = {}
+    _parameters = self.attributes['parameters']
+    dt_col = _parameters.get('columns', {}).get('datetime', None)
+    dt_typ = _parameters.get('dtypes', {}).get(dt_col, None) if dt_col else None
+    if dt_col and not dt_typ:
+        if 'dtypes' not in _parameters:
+            self.attributes['parameters']['dtypes'] = {}
+        self.attributes['parameters']['dtypes'][dt_col] = 'datetime'
     return self.attributes['parameters']
 
 
@@ -260,6 +270,25 @@ def autoincrement(self, _autoincrement: bool) -> None:
     self.parameters['autoincrement'] = _autoincrement
 
 
+@property
+def tzinfo(self) -> Union[None, timezone]:
+    """
+    Return `timezone.utc` if the pipe is timezone-aware.
+    """
+    dt_col = self.columns.get('datetime', None)
+    if not dt_col:
+        return None
+
+    dt_typ = str(self.dtypes.get(dt_col, 'datetime64[ns, UTC]'))
+    if 'utc' in dt_typ.lower() or dt_typ == 'datetime':
+        return timezone.utc
+
+    if dt_typ == 'datetime64[ns]':
+        return None
+
+    return None
+
+
 def get_columns(self, *args: str, error: bool = False) -> Union[str, Tuple[str]]:
     """
     Check if the requested columns are defined.

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/core/Pipe/_dtypes.py

@@ -30,6 +30,7 @@ def enforce_dtypes(
     from meerschaum.utils.warnings import warn
     from meerschaum.utils.debug import dprint
     from meerschaum.utils.dataframe import parse_df_datetimes, enforce_dtypes as _enforce_dtypes
+    from meerschaum.utils.dtypes import are_dtypes_equal
     from meerschaum.utils.packages import import_pandas
     pd = import_pandas(debug=debug)
     if df is None:
@@ -51,6 +52,7 @@ def enforce_dtypes(
                     for col, dtype in pipe_dtypes.items()
                     if 'datetime' not in str(dtype)
                 ],
+                strip_timezone=(self.tzinfo is None),
                 chunksize=chunksize,
                 debug=debug,
             )
@@ -60,8 +62,9 @@ def enforce_dtypes(
                 ignore_cols=[
                     col
                     for col, dtype in pipe_dtypes.items()
-                    if 'datetime' not in str(dtype)
+                    if not are_dtypes_equal(str(dtype), 'datetime')
                 ],
+                strip_timezone=(self.tzinfo is None),
                 chunksize=chunksize,
                 debug=debug,
             )
@@ -77,7 +80,14 @@ def enforce_dtypes(
             )
         return df
 
-    return _enforce_dtypes(df, pipe_dtypes, safe_copy=safe_copy, debug=debug)
+    return _enforce_dtypes(
+        df,
+        pipe_dtypes,
+        safe_copy=safe_copy,
+        strip_timezone=(self.tzinfo is None),
+        coerce_timezone=True,
+        debug=debug,
+    )
 
 
 def infer_dtypes(self, persist: bool = False, debug: bool = False) -> Dict[str, Any]:

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/core/Pipe/_sync.py

@@ -933,7 +933,11 @@ def _persist_new_numeric_columns(self, df, debug: bool = False) -> SuccessTuple:
     if not new_numeric_cols:
         return True, "Success"
 
+    self._attributes_sync_time = None
+    dt_col = self.columns.get('datetime', None)
     dtypes = self.parameters.get('dtypes', {})
+    if dt_col not in dtypes:
+        dtypes[dt_col] = 'datetime'
     dtypes.update({col: 'numeric' for col in numeric_cols})
     self.parameters['dtypes'] = dtypes
     if not self.temporary:
@@ -957,7 +961,11 @@ def _persist_new_uuid_columns(self, df, debug: bool = False) -> SuccessTuple:
     if not new_uuid_cols:
         return True, "Success"
 
+    self._attributes_sync_time = None
+    dt_col = self.columns.get('datetime', None)
     dtypes = self.parameters.get('dtypes', {})
+    if dt_col not in dtypes:
+        dtypes[dt_col] = 'datetime'
     dtypes.update({col: 'uuid' for col in uuid_cols})
     self.parameters['dtypes'] = dtypes
     if not self.temporary:
@@ -981,7 +989,11 @@ def _persist_new_json_columns(self, df, debug: bool = False) -> SuccessTuple:
     if not new_json_cols:
         return True, "Success"
 
+    self._attributes_sync_time = None
+    dt_col = self.columns.get('datetime', None)
     dtypes = self.parameters.get('dtypes', {})
+    if dt_col not in dtypes:
+        dtypes[dt_col] = 'datetime'
     dtypes.update({col: 'json' for col in json_cols})
     self.parameters['dtypes'] = dtypes
 

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/utils/dataframe.py

@@ -689,6 +689,7 @@ def enforce_dtypes(
     safe_copy: bool = True,
     coerce_numeric: bool = True,
     coerce_timezone: bool = True,
+    strip_timezone: bool = False,
     debug: bool = False,
 ) -> 'pd.DataFrame':
     """
@@ -713,6 +714,10 @@ def enforce_dtypes(
     coerce_timezone: bool, default True
         If `True`, convert datetimes to UTC.
 
+    strip_timezone: bool, default False
+        If `coerce_timezone` and `strip_timezone` are `True`,
+        remove timezone information from datetimes.
+
     debug: bool, default False
         Verbosity toggle.
 
@@ -817,7 +822,7 @@ def enforce_dtypes(
             dprint(f"Checking for datetime conversion: {datetime_cols}")
         for col in datetime_cols:
             if col in df.columns:
-                df[col] = _coerce_timezone(df[col])
+                df[col] = _coerce_timezone(df[col], strip_utc=strip_timezone)
 
     df_dtypes = {c: str(t) for c, t in df.dtypes.items()}
     if are_dtypes_equal(df_dtypes, pipe_pandas_dtypes):

{meerschaum-2.6.0 → meerschaum-2.6.1}/meerschaum/utils/dtypes/sql.py

@@ -304,6 +304,19 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
         'cockroachdb': 'Float',
         'default': 'Float',
     },
+    'datetime': {
+        'timescaledb': 'DateTime(timezone=True)',
+        'postgresql': 'DateTime(timezone=True)',
+        'mariadb': 'DateTime(timezone=True)',
+        'mysql': 'DateTime(timezone=True)',
+        'mssql': 'sqlalchemy.dialects.mssql.DATETIMEOFFSET',
+        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP(timezone=True)',
+        'sqlite': 'DateTime(timezone=True)',
+        'duckdb': 'DateTime(timezone=True)',
+        'citus': 'DateTime(timezone=True)',
+        'cockroachdb': 'DateTime(timezone=True)',
+        'default': 'DateTime(timezone=True)',
+    },
     'datetime64[ns]': {
         'timescaledb': 'DateTime',
         'postgresql': 'DateTime',
@@ -489,7 +502,7 @@ def get_db_type_from_pd_type(
     """
     from meerschaum.utils.warnings import warn
     from meerschaum.utils.packages import attempt_import
-    from meerschaum.utils.dtypes import are_dtypes_equal, MRSM_PD_DTYPES
+    from meerschaum.utils.dtypes import are_dtypes_equal
     from meerschaum.utils.misc import parse_arguments_str
     sqlalchemy_types = attempt_import('sqlalchemy.types')