Add APIClientProvider to widget

optuna · Jul 10, 2024 · 52f7d62 · 52f7d62
1 parent 7c0b18f
commit 52f7d62
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/jupyterlab/src/widget.tsx b/jupyterlab/src/widget.tsx
@@ -1,6 +1,10 @@
 import { ReactWidget } from '@jupyterlab/ui-components';
 import React from 'react';
 import { JupyterLabEntrypoint } from './components/JupyterLabEntrypoint';
+import { JupyterlabAPIClient } from './apiClient';
+import { APIClientProvider } from '@optuna/optuna-dashboard';
+
+const jupyterlabAPIClient = new JupyterlabAPIClient();
 
 export class OptunaDashboardWidget extends ReactWidget {
   constructor() {
@@ -9,6 +13,10 @@ export class OptunaDashboardWidget extends ReactWidget {
   }
 
   render(): JSX.Element {
-    return <JupyterLabEntrypoint />;
+    return (
+      <APIClientProvider apiClient={jupyterlabAPIClient}>
+        <JupyterLabEntrypoint />
+      </APIClientProvider>
+    );
   }
 }