From 7fe31a56fe2d9f4972a7dc28cab5500f3e8cf159 Mon Sep 17 00:00:00 2001 From: Jakub Walczak Date: Fri, 19 Jan 2024 10:39:58 +0100 Subject: [PATCH] Fix statis typing for streamz DataFrame --- drivers/intake_geokube/iot/driver.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/drivers/intake_geokube/iot/driver.py b/drivers/intake_geokube/iot/driver.py index 93c52cd..ce5fca0 100644 --- 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]