def setUpClass(cls):
super(TestRKernelLocal, cls).setUpClass()
print('\nStarting R kernel using {} kernelspec'.format(cls.KERNELSPEC))
# initialize environment
cls.gatewayClient = GatewayClient()
cls.kernel = cls.gatewayClient.start_kernel(cls.KERNELSPEC)
def setUpClass(cls):
super().setUpClass()
print(f"\nStarting Python kernel using {cls.KERNELSPEC} kernelspec")
# initialize environment
cls.gatewayClient = GatewayClient()
cls.kernel = cls.gatewayClient.start_kernel(cls.KERNELSPEC)
def setUpClass(cls):
super(TestPythonKernelCluster, cls).setUpClass()
print('>>>')
print('Starting Python kernel using {} kernelspec'.format(cls.KERNELSPEC))
# initialize environment
cls.gatewayClient = GatewayClient()
cls.kernel = cls.gatewayClient.start_kernel(cls.KERNELSPEC)
def execute_task(self, task):
# start notebook
print('')
print('Start notebook execution...')
print('Starting kernel...')
launcher = GatewayClient(task['endpoint'])
kernel = launcher.start_kernel(task['kernelspec'])
time.sleep(10)
# execute all cells
try:
print('reading notebook contents')
notebook = nbformat.reads(json.dumps(task['notebook']),
as_version=4)
print('Starting cell execution')
for cell in notebook.cells:
print('Executing cell\n{}'.format(cell.source))
response = kernel.execute(cell.source)
print('Response\n{}'.format(response))
outputs = []
outputs.append(response)
cell['outputs'] = outputs
except BaseException as base:
print('Error executing notebook cells: {}'.format(base))
finally:
print('Starting kernel shutdown')
# shutdown notebook
launcher.shutdown_kernel(kernel)
print('Notebook execution done')
print('')
from flask_sse import sse
from flask import Blueprint
from copy import deepcopy
from mongoengine import DoesNotExist
from nbformat import v4 as nbf
from enterprise_gateway.client.gateway_client import GatewayClient
from mpcontribs.api.core import SwaggerView
from mpcontribs.api.notebooks.document import Notebooks
from mpcontribs.api.contributions.document import Contributions
from mpcontribs.api.tables.document import Tables
from mpcontribs.api.structures.document import Structures
templates = os.path.join(os.path.dirname(flask_mongorest.__file__),
"templates")
notebooks = Blueprint("notebooks", __name__, template_folder=templates)
client = GatewayClient()
seed_nb = nbf.new_notebook()
seed_nb["cells"] = [
nbf.new_code_cell("\n".join([
"from mpcontribs.client import load_client",
"from mpcontribs.io.core.components.hdata import HierarchicalData",
"from mpcontribs.io.core.components.tdata import Table # DataFrame with Backgrid IPython Display",
"from mpcontribs.io.core.components.gdata import Plot # Plotly interactive graph",
"from pymatgen import Structure",
]))
]
class NotebooksResource(Resource):
document = Notebooks
filters = {"is_public": [ops.Boolean]}
def setUpClass(cls):
super(TestAuthorization, cls).setUpClass()
print('>>>')
# initialize environment
cls.gateway_client = GatewayClient()
def setUpClass(cls):
super().setUpClass()
print(">>>")
# initialize environment
cls.gateway_client = GatewayClient()