Skip to content

Commit d487329

Browse files
remove type error that is now picked up by mypy
1 parent 4d22c6e commit d487329

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/databricks/sql/auth/retry.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def __init__(
139139

140140
@classmethod
141141
def __private_init__(
142-
cls, retry_start_time: float, command_type: Optional[CommandType], **init_kwargs
142+
cls, retry_start_time: float, command_type: CommandType, **init_kwargs
143143
):
144144
"""
145145
Returns a new instance of DatabricksRetryPolicy with the _retry_start_time and _command_type
@@ -238,7 +238,7 @@ def new(
238238
)
239239

240240
@property
241-
def command_type(self) -> Optional[CommandType]:
241+
def command_type(self) -> CommandType:
242242
return self._command_type
243243

244244
@command_type.setter

0 commit comments

Comments
 (0)