Remove statis typing for streamz dataframe

CMCC-Foundation · Jan 22, 2024 · 7ce6244 · 7ce6244
1 parent 5a70d03
commit 7ce6244
Showing 1 changed file with 2 additions and 2 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.DataFrame:
+    def read(self):
         """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.DataFrame:
+    def process(self, query: GeoQuery):
         """Process IoT data with the passed query.
 
         Parameters