ignore missing type stubs

dune_client/api/extensions.py

@@ -77,7 +77,7 @@ def run_query_dataframe(
         This is a convenience method that uses run_query_csv() + pandas.read_csv() underneath
         """
         try:
-            import pandas  # type: ignore # pylint: disable=import-outside-toplevel
+            import pandas  # pylint: disable=import-outside-toplevel
         except ImportError as exc:
             raise ImportError(
                 "dependency failure, pandas is required but missing"

dune_client/client_async.py

@@ -271,7 +271,7 @@ async def refresh_into_dataframe(
         This is a convenience method that uses refresh_csv underneath
         """
         try:
-            import pandas  # type: ignore # pylint: disable=import-outside-toplevel
+            import pandas  # pylint: disable=import-outside-toplevel
         except ImportError as exc:
             raise ImportError(
                 "dependency failure, pandas is required but missing"

dune_client/viz/graphs.py

@@ -3,9 +3,12 @@
 """
 
 from typing import Dict, Union
-import plotly.graph_objects as go
-import colorlover as cl
+
+# https://github.com/plotly/colorlover/issues/35
+import colorlover as cl  # type: ignore[import]
 import pandas as pd
+import plotly.graph_objects as go  # type: ignore[import]
+from plotly.graph_objs import Figure  # type: ignore[import]
 
 
 # function to create Sankey diagram
@@ -15,7 +18,7 @@ def create_sankey(
     columns: Dict[str, str],
     viz_config: Dict[str, Union[int, float]],
     title: str = "unnamed",
-):
+) -> Figure:
     """
     Creates a Sankey diagram based on input query_result
     , which must contain source, target, value columns

requirements/dev.txt

@@ -1,6 +1,7 @@
 -r prod.txt
 black>=23.7.0
 pandas>=1.0.0
+pandas-stubs>=1.0.0
 pylint>=2.17.5
 pytest>=7.4.1
 python-dotenv>=1.0.0