@@ -37,7 +37,7 @@ def __init__(
3737 on_display_data : Optional [Callable [[Dict [MIMEType , str ]], None ]] = None ,
3838 ):
3939 self .partial_result = Execution ()
40- self .result = Future ()
40+ self .execution = Future ()
4141 self .on_stdout = on_stdout
4242 self .on_stderr = on_stderr
4343 self .on_display_data = on_display_data
@@ -148,7 +148,7 @@ def send_execution_message(
148148 def get_result (
149149 self , message_id : str , timeout : Optional [float ] = TIMEOUT
150150 ) -> Execution :
151- result = self ._cells [message_id ].result .result (timeout = timeout )
151+ result = self ._cells [message_id ].execution .result (timeout = timeout )
152152 logger .debug (f"Got result for message: { message_id } " )
153153 del self ._cells [message_id ]
154154 return result
@@ -169,19 +169,19 @@ def _receive_message(self, data: dict):
169169 if not cell :
170170 return
171171
172- result = cell .partial_result
172+ execution = cell .partial_result
173173
174174 if data ["msg_type" ] == "error" :
175175 logger .debug (f"Cell { parent_msg_ig } finished execution with error" )
176- result .error = Error (
176+ execution .error = Error (
177177 name = data ["content" ]["ename" ],
178178 value = data ["content" ]["evalue" ],
179179 traceback_raw = data ["content" ]["traceback" ],
180180 )
181181
182182 elif data ["msg_type" ] == "stream" :
183183 if data ["content" ]["name" ] == "stdout" :
184- result .logs .stdout .append (data ["content" ]["text" ])
184+ execution .logs .stdout .append (data ["content" ]["text" ])
185185 if cell .on_stdout :
186186 cell .on_stdout (
187187 ProcessMessage (
@@ -191,7 +191,7 @@ def _receive_message(self, data: dict):
191191 )
192192
193193 elif data ["content" ]["name" ] == "stderr" :
194- result .logs .stderr .append (data ["content" ]["text" ])
194+ execution .logs .stderr .append (data ["content" ]["text" ])
195195 if cell .on_stderr :
196196 cell .on_stderr (
197197 ProcessMessage (
@@ -202,34 +202,33 @@ def _receive_message(self, data: dict):
202202 )
203203
204204 elif data ["msg_type" ] in "display_data" :
205- result .results .append (
206- Result (is_main_result = False , data = data ["content" ]["data" ])
207- )
205+ result = Result (is_main_result = False , data = data ["content" ]["data" ])
206+ execution .results .append (result )
208207 if cell .on_display_data :
209- cell .on_display_data (data [ "content" ][ "data" ] )
208+ cell .on_display_data (result )
210209 elif data ["msg_type" ] == "execute_result" :
211- result .results .append (
210+ execution .results .append (
212211 Result (is_main_result = True , data = data ["content" ]["data" ])
213212 )
214213 elif data ["msg_type" ] == "status" :
215214 if data ["content" ]["execution_state" ] == "idle" :
216215 if cell .input_accepted :
217216 logger .debug (f"Cell { parent_msg_ig } finished execution" )
218- cell .result .set_result (result )
217+ cell .execution .set_result (execution )
219218
220219 elif data ["content" ]["execution_state" ] == "error" :
221220 logger .debug (f"Cell { parent_msg_ig } finished execution with error" )
222- result .error = Error (
221+ execution .error = Error (
223222 name = data ["content" ]["ename" ],
224223 value = data ["content" ]["evalue" ],
225224 traceback_raw = data ["content" ]["traceback" ],
226225 )
227- cell .result .set_result (result )
226+ cell .execution .set_result (execution )
228227
229228 elif data ["msg_type" ] == "execute_reply" :
230229 if data ["content" ]["status" ] == "error" :
231230 logger .debug (f"Cell { parent_msg_ig } finished execution with error" )
232- result .error = Error (
231+ execution .error = Error (
233232 name = data ["content" ]["ename" ],
234233 value = data ["content" ]["evalue" ],
235234 traceback_raw = data ["content" ]["traceback" ],
0 commit comments