Skip to content

Commit 6962fc0

Browse files
committed
fix state check to be logical
1 parent 45e43ea commit 6962fc0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytrickle/frame_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def attach_state(self, state: StreamState) -> None:
6767
async def ensure_model_loaded(self, **kwargs):
6868
"""Thread-safe wrapper that ensures model is loaded exactly once."""
6969
async with self._model_load_lock:
70-
if not self.state.get_state() == PipelineState.LOADING:
70+
if not self.state.get_state() == PipelineState.IDLE:
7171
await self.load_model(**kwargs)
7272
self.state.set_startup_complete()
7373
logger.debug(f"Model loaded for {self.__class__.__name__}")

0 commit comments

Comments
 (0)