Skip to content

Commit 188a659

Browse files
author
marwan37
committed
fix type annotation for wrapper cache in lsp_zenml.py
1 parent ba23884 commit 188a659

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

bundled/tool/lsp_zenml.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import subprocess
2323
import sys
2424
from functools import wraps
25+
from typing import Any, Dict
2526

2627
import lsprotocol.types as lsp
2728
from constants import IS_ZENML_INSTALLED, MIN_ZENML_VERSION, TOOL_MODULE_NAME
@@ -135,7 +136,7 @@ def wrapper(*args, **kwargs):
135136
# Cache wrapper instances to avoid repeated getattr calls
136137
if wrapper_name:
137138
if not hasattr(self, "_wrapper_cache"):
138-
self._wrapper_cache = {}
139+
self._wrapper_cache: Dict[str, Any] = {}
139140

140141
if wrapper_name not in self._wrapper_cache:
141142
wrapper_instance = getattr(self.zenml_client, wrapper_name, None)

0 commit comments

Comments
 (0)