From 7853dc9da71d1d9cbc769ee55fb7f3b7a4f81463 Mon Sep 17 00:00:00 2001 From: Benedikt Bartscher Date: Sat, 21 Dec 2024 17:47:16 +0100 Subject: [PATCH] add type ignores.. :/ --- benchmarks/benchmark_pickle.py | 4 ++-- reflex/state.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/benchmarks/benchmark_pickle.py b/benchmarks/benchmark_pickle.py index 7773d3b2e..c282874b8 100644 --- a/benchmarks/benchmark_pickle.py +++ b/benchmarks/benchmark_pickle.py @@ -109,10 +109,10 @@ def test_pickle( pytest.skip("Blosc is not available.") 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: - return pickle.loads(decompress(data)) + return pickle.loads(decompress(data)) # type: ignore else: diff --git a/reflex/state.py b/reflex/state.py index 59c84d50e..3fb5d8dad 100644 --- a/reflex/state.py +++ b/reflex/state.py @@ -2256,7 +2256,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow): prefix = STATE_COMPRESSED else: prefix = STATE_NOT_COMPRESSED - payload = prefix + payload + payload = prefix + payload # type: ignore return payload @@ -2288,7 +2288,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow): if is_compressed: from blosc2 import decompress - data = decompress(data) + data = decompress(data) # type: ignore data = pickle.loads(data) # type: ignore elif fp is not None and data is None: if environment.REFLEX_COMPRESS_STATE.get(): @@ -2305,7 +2305,7 @@ class BaseState(Base, ABC, extra=pydantic.Extra.allow): fp = BytesIO() 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) else: