Add support for custom indexes for query in the DataFrameClient (influxdata#785)

rolincova · rolincova · commit 9a110a1abda2 · 2020-06-01T09:57:12.000+02:00
diff --git a/influxdb/_dataframe_client.py b/influxdb/_dataframe_client.py
@@ -8,7 +8,6 @@
 
 import math
 from collections import defaultdict
-from typing import List
 
 import pandas as pd
 import numpy as np
@@ -154,7 +153,7 @@ def query(self,
               chunk_size=0,
               method="GET",
               dropna=True,
-              data_frame_index: List[str] = None):
+              data_frame_index=None):
         """
         Query data into a DataFrame.
 
@@ -205,7 +204,7 @@ def query(self,
         else:
             return results
 
-    def _to_dataframe(self, rs, dropna=True, data_frame_index: List[str] = None):
+    def _to_dataframe(self, rs, dropna=True, data_frame_index=None):
         result = defaultdict(list)
         if isinstance(rs, list):
             return map(self._to_dataframe, rs,
