Skip to content

Commit aa06b72

Browse files
[licensing refactor] remove frozendict dependency, use types.MappingProxyType instead (#469)
* frozendict -> MappingProxyType Signed-off-by: Brian Dellabetta <[email protected]> * style/test fixes Signed-off-by: Brian Dellabetta <[email protected]> --------- Signed-off-by: Brian Dellabetta <[email protected]>
1 parent c649159 commit aa06b72

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def _setup_packages() -> List:
8888
)
8989

9090
def _setup_install_requires() -> List:
91-
return ["torch>=1.7.0", "transformers", "pydantic>=2.0", "frozendict", "loguru"]
91+
return ["torch>=1.7.0", "transformers", "pydantic>=2.0", "loguru"]
9292

9393
def _setup_extras() -> Dict:
9494
return {

src/compressed_tensors/utils/helpers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,11 @@
1515
import contextlib
1616
import warnings
1717
from functools import wraps
18+
from types import MappingProxyType
1819
from typing import TYPE_CHECKING, Any, Callable, Dict, List, Mapping, Optional, TypeVar
1920

2021
import numpy
2122
import torch
22-
from frozendict import frozendict
2323
from transformers import AutoConfig
2424

2525

@@ -379,7 +379,7 @@ class ParameterizedDefaultDict(dict):
379379

380380
def __init__(self, default_factory: Callable[[Any], Any]):
381381
self.default_factory = default_factory
382-
self._factory_kwargs = frozendict()
382+
self._factory_kwargs = MappingProxyType({})
383383

384384
def __missing__(self, key: Any) -> Any:
385385
if isinstance(key, tuple):
@@ -389,7 +389,7 @@ def __missing__(self, key: Any) -> Any:
389389
self[key] = value
390390
return value
391391

392-
def get(self, *args, factory_kwargs: Mapping = frozendict()) -> Any:
392+
def get(self, *args, factory_kwargs: Mapping = MappingProxyType({})) -> Any:
393393
"""
394394
Similar to `__getitem__`, but allows passing kwargs to factory function
395395

0 commit comments

Comments
 (0)