Skip to content

Commit

Permalink
after exec function
Browse files Browse the repository at this point in the history
  • Loading branch information
Cole Blanchard authored and Cole Blanchard committed Jul 17, 2024
1 parent f4ce2b5 commit 65aa9a6
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions src/askem_beaker/contexts/model_configuration/context.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,6 +119,15 @@ async def load_config(self):
print(f"Running command:\n-------\n{command}\n---------")
await self.execute(command)

async def post_execute(self, message):
try:
content = (await self.evaluate(self.get_code("get_config")))["return"]
self.beaker_kernel.send_response(
"iopub", "model_configuration_preview", content, parent_header=message.parent_header
)
except Exception as e:
raise

@intercept()
async def save_model_config_request(self, message):
'''
Expand Down

0 comments on commit 65aa9a6

Please sign in to comment.