fix mypy checks due to upgrade

PrefectHQ · Mar 9, 2020 · 69bef89 · 69bef89
1 parent 16e7940
commit 69bef89
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 6 deletions.
diff --git a/src/prefect/agent/agent.py b/src/prefect/agent/agent.py
@@ -381,7 +381,7 @@ def query_flow_runs(self, tenant_id: str) -> list:
                                     "state": {"_eq": "Running"},
                                     "task_runs": {
                                         "state_start_time": {
-                                            "_lte": str(now.subtract(seconds=3))
+                                            "_lte": str(now.subtract(seconds=3))  # type: ignore
                                         }
                                     },
                                 },
@@ -401,7 +401,7 @@ def query_flow_runs(self, tenant_id: str) -> list:
                         {
                             "where": {
                                 "state_start_time": {
-                                    "_lte": str(now.subtract(seconds=3))
+                                    "_lte": str(now.subtract(seconds=3))  # type: ignore
                                 }
                             }
                         },

diff --git a/src/prefect/agent/kubernetes/agent.py b/src/prefect/agent/kubernetes/agent.py
@@ -21,7 +21,7 @@ def check_heartbeat() -> None:
     """
     Check the agent's heartbeat by verifying heartbeat file has been recently modified
     """
-    current_timestamp = pendulum.now().timestamp()
+    current_timestamp = pendulum.now().timestamp()  # type: ignore
     last_modified_timestamp = path.getmtime("{}/heartbeat".format(AGENT_DIRECTORY))
 
     # If file has not been modified in the last 40 seconds then raise an exit code of 1

diff --git a/src/prefect/engine/result_handlers/azure_result_handler.py b/src/prefect/engine/result_handlers/azure_result_handler.py
@@ -99,7 +99,7 @@ def write(self, result: Any) -> str:
         Returns:
             - str: the Blob URI
         """
-        date = pendulum.now("utc").format("Y/M/D")
+        date = pendulum.now("utc").format("Y/M/D")  # type: ignore
         uri = "{date}/{uuid}.prefect_result".format(date=date, uuid=uuid.uuid4())
         self.logger.debug("Starting to upload result to {}...".format(uri))
 

diff --git a/src/prefect/engine/result_handlers/gcs_result_handler.py b/src/prefect/engine/result_handlers/gcs_result_handler.py
@@ -77,7 +77,7 @@ def write(self, result: Any) -> str:
         Returns:
             - str: the GCS URI
         """
-        date = pendulum.now("utc").format("Y/M/D")
+        date = pendulum.now("utc").format("Y/M/D")  # type: ignore
         uri = "{date}/{uuid}.prefect_result".format(date=date, uuid=uuid.uuid4())
         self.logger.debug("Starting to upload result to {}...".format(uri))
         binary_data = base64.b64encode(cloudpickle.dumps(result)).decode()

diff --git a/src/prefect/engine/result_handlers/s3_result_handler.py b/src/prefect/engine/result_handlers/s3_result_handler.py
@@ -91,7 +91,7 @@ def write(self, result: Any) -> str:
         Returns:
             - str: the S3 URI
         """
-        date = pendulum.now("utc").format("Y/M/D")
+        date = pendulum.now("utc").format("Y/M/D")  # type: ignore
         uri = "{date}/{uuid}.prefect_result".format(date=date, uuid=uuid.uuid4())
         self.logger.debug("Starting to upload result to {}...".format(uri))
 

diff --git a/src/prefect/schedules/clocks.py b/src/prefect/schedules/clocks.py
@@ -255,6 +255,7 @@ def events(self, after: datetime = None) -> Iterable[ClockEvent]:
         assert isinstance(after, datetime)  # mypy assertion
         after = pendulum.instance(after)
         assert isinstance(after, pendulum.DateTime)  # mypy assertion
+        assert isinstance(after.tz, pendulum.tz._Timezone)  # mypy assertion
 
         # croniter's DST logic interferes with all other datetime libraries except pytz
         after_localized = pytz.timezone(after.tz.name).localize(