Skip to content

Commit f1ef74f

Browse files
authored
Merge pull request #859 from TimDettmers/younesbelkada-patch-1
Fix typehint error
2 parents fab00e2 + 67adb47 commit f1ef74f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bitsandbytes/functional.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
import numpy as np
1414

1515
from functools import reduce # Required in Python 3
16-
from typing import Tuple, Any
16+
from typing import Tuple, Any, Dict
1717
from torch import Tensor
1818
from bitsandbytes.utils import pack_dict_to_tensor, unpack_tensor_to_dict
1919

@@ -600,7 +600,7 @@ def __get_item__(self, idx):
600600
return list_repr[idx]
601601

602602
@classmethod
603-
def from_dict(cls, qs_dict: dict[str, Any], device: torch.device) -> 'QuantState':
603+
def from_dict(cls, qs_dict: Dict[str, Any], device: torch.device) -> 'QuantState':
604604
"""
605605
unpacks components of state_dict into QuantState
606606
where necessary, convert into strings, torch.dtype, ints, etc.

0 commit comments

Comments
 (0)