laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add type ignores.. :/

Browse Source
This commit is contained in:
Benedikt Bartscher 2024-12-21 17:47:16 +01:00
parent e2a58c2d6f
commit 7853dc9da7
No known key found for this signature in database
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
benchmarks/benchmark_pickle.py
View File

@ -109,10 +109,10 @@ def test_pickle(
pytest.skip("Blosc is not available.") pytest.skip("Blosc is not available.")
def dump(obj: State) -> bytes: def dump(obj: State) -> bytes:
return compress(pickle.dumps(obj, protocol=protocol)) return compress(pickle.dumps(obj, protocol=protocol)) # type: ignore
def load(data: bytes) -> State: def load(data: bytes) -> State:
return pickle.loads(decompress(data)) return pickle.loads(decompress(data)) # type: ignore
else: else:

6
reflex/state.py
View File

@ -2256,7 +2256,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
prefix = STATE_COMPRESSED prefix = STATE_COMPRESSED
else: else:
prefix = STATE_NOT_COMPRESSED prefix = STATE_NOT_COMPRESSED
payload = prefix + payload payload = prefix + payload # type: ignore
return payload return payload
@ -2288,7 +2288,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
if is_compressed: if is_compressed:
from blosc2 import decompress from blosc2 import decompress
data = decompress(data) data = decompress(data) # type: ignore
data = pickle.loads(data) # type: ignore data = pickle.loads(data) # type: ignore
elif fp is not None and data is None: elif fp is not None and data is None:
if environment.REFLEX_COMPRESS_STATE.get(): if environment.REFLEX_COMPRESS_STATE.get():
@ -2305,7 +2305,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow):
fp = BytesIO() fp = BytesIO()
for chunk_index in range(schunk.nchunks): for chunk_index in range(schunk.nchunks):
fp.write(schunk.decompress_chunk(chunk_index)) fp.write(schunk.decompress_chunk(chunk_index)) # type: ignore
data = pickle.load(fp) data = pickle.load(fp)
else: else: