From 5d0b9eb26b2b5c951818932bbeb47d040f11df81 Mon Sep 17 00:00:00 2001
From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com>
Date: Mon, 7 Oct 2024 09:04:02 +0200
Subject: [PATCH] lint neptune after 1.12.0 release (#3055)

---
 .../integrations/neptune/experiment_trackers/run_state.py     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/zenml/integrations/neptune/experiment_trackers/run_state.py b/src/zenml/integrations/neptune/experiment_trackers/run_state.py
index 56080681355..145bc464fab 100644
--- a/src/zenml/integrations/neptune/experiment_trackers/run_state.py
+++ b/src/zenml/integrations/neptune/experiment_trackers/run_state.py
@@ -16,7 +16,7 @@
 from hashlib import md5
 from typing import TYPE_CHECKING, Any, List, Optional
 
-import neptune  # type: ignore[import-untyped]
+import neptune
 
 import zenml
 from zenml.client import Client
@@ -38,7 +38,7 @@ class RunProvider(metaclass=SingletonMetaClass):
 
     def __init__(self) -> None:
         """Initialize RunProvider. Called with no arguments."""
-        self._active_run = None
+        self._active_run: Optional["Run"] = None
         self._project: Optional[str]
         self._run_name: Optional[str]
         self._token: Optional[str]