Skip to content

Commit 4d0f45c

Browse files
committed
some more optimisations
Signed-off-by: Praneeth Bedapudi <[email protected]>
1 parent b8b49d3 commit 4d0f45c

File tree

4 files changed

+280
-145
lines changed

4 files changed

+280
-145
lines changed

fastdeploy/_infer.py

+1
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ def add_to_infer_queue(
189189
"-1.predicted_at": 0,
190190
"last_predictor_sequence": -1,
191191
"last_predictor_success": True,
192+
"timedout_in_queue": None,
192193
}
193194
}
194195
)

fastdeploy/_loop.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,7 @@ def start_loop(
197197
"-1.predicted_at": 0,
198198
"-1.received_at": {"$lt": time.time() - timeout_time},
199199
"timedout_in_queue": {"$ne": True},
200+
"last_predictor_sequence": {"$ne": _utils.LAST_PREDICTOR_SEQUENCE},
200201
},
201202
update={"timedout_in_queue": True},
202203
select_keys=[],
@@ -243,8 +244,6 @@ def start_loop(
243244
f"Updated results predictor {predictor_sequence}: {list(unique_id_wise_results)}"
244245
)
245246

246-
last_batch_collection_started_at = time.time()
247-
248247

249248
if __name__ == "__main__":
250249
import sys

fastdeploy/_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
logging.basicConfig(
44
format="%(asctime)s,%(msecs)d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s",
55
datefmt="%Y-%m-%d:%H:%M:%S",
6-
level=logging.DEBUG,
6+
level=logging.INFO,
77
)
88

99
logger = logging.getLogger(__name__)

0 commit comments

Comments
 (0)