File tree Expand file tree Collapse file tree 1 file changed +10
-13
lines changed
Expand file tree Collapse file tree 1 file changed +10
-13
lines changed Original file line number Diff line number Diff line change @@ -310,19 +310,16 @@ def get_key(self) -> str:
310310 return f"{ self .request_id } _{ self .start_kv_index } _{ self .end_kv_index } "
311311
312312 def to_str (self ):
313- if self .mem_indexes is not None :
314- obj : NIXLChunckedTransTask = copy .copy (self )
315- obj .mem_indexes = None
316- if obj .decode_agent_metadata is not None :
317- obj .decode_agent_metadata = b"xxx"
318- if obj .prefill_agent_metadata is not None :
319- obj .prefill_agent_metadata = b"xxx"
320- if obj .decode_page_reg_desc is not None :
321- obj .decode_page_reg_desc = b"xxx"
322- if obj .prefill_page_reg_desc is not None :
323- obj .prefill_page_reg_desc = b"xxx"
324- else :
325- obj = self
313+ obj : NIXLChunckedTransTask = copy .copy (self )
314+ obj .mem_indexes = None
315+ if obj .decode_agent_metadata is not None :
316+ obj .decode_agent_metadata = b"xxx"
317+ if obj .prefill_agent_metadata is not None :
318+ obj .prefill_agent_metadata = b"xxx"
319+ if obj .decode_page_reg_desc is not None :
320+ obj .decode_page_reg_desc = b"xxx"
321+ if obj .prefill_page_reg_desc is not None :
322+ obj .prefill_page_reg_desc = b"xxx"
326323 return obj .__str__ ()
327324
328325 def transfer_kv_num (self ):
You can’t perform that action at this time.
0 commit comments