Fix statis typing for streamz DataFrame

CMCC-Foundation · Jan 19, 2024 · 7fe31a5 · 7fe31a5
1 parent 1836d60
commit 7fe31a5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/drivers/intake_geokube/iot/driver.py b/drivers/intake_geokube/iot/driver.py
@@ -100,7 +100,7 @@ def _get_schema(self):
             self.stream.start()
         return {"stream": str(self.stream)}
 
-    def read(self) -> streamz.dataframe.core.DataFrame:
+    def read(self) -> streamz.dataframe.DataFrame:
         """Read IoT data."""
         self.log.info("reading stream...")
         self._get_schema()
@@ -113,7 +113,7 @@ def load(self) -> NoReturn:
             "loading entire product is not supported for IoT data"
         )
 
-    def process(self, query: GeoQuery) -> streamz.dataframe.core.DataFrame:
+    def process(self, query: GeoQuery) -> streamz.dataframe.DataFrame:
         """Process IoT data with the passed query.
 
         Parameters
@@ -123,7 +123,7 @@ def process(self, query: GeoQuery) -> streamz.dataframe.core.DataFrame:
 
         Returns
         -------
-        stream  : streamz.dataframe.core.DataFrame
+        stream  : streamz.dataframe.DataFrame
             A DataFrame object with streamed content
         """
         df = d[0]