use pandas stubs

whylabs · Dec 11, 2024 · 6cf62bb · 6cf62bb
1 parent f2900b1
commit 6cf62bb
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/python/whylogs/api/logger/experimental/logger/actor/data_logger.py b/python/whylogs/api/logger/experimental/logger/actor/data_logger.py
@@ -1,11 +1,7 @@
 from abc import abstractmethod
 from typing import Any, Dict, Generic, List, Optional, TypeVar, Union
 
-# TODO: stubs?
-try:
-    import pandas as pd  # type: ignore
-except ImportError:
-    pd: Any = None  # type: ignore
+from whylogs.core.stubs import pd
 
 
 # TODO add strong typing here. whylogs takes pretty much anything in log() so we don't actually