clean up

Author: Artmann

src/kernels/deepnote/deepnoteController.ts

@@ -1,32 +1,13 @@
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT License.
-
 import {
     CancellationToken,
     NotebookCell,
     NotebookCellExecution,
     NotebookCellOutput,
-    NotebookCellOutputItem,
-    NotebookController
+    NotebookCellOutputItem
 } from 'vscode';
 
 import { KernelController } from '../kernelController';
 
-/**
- * DeepnoteController extends KernelController to intercept cell execution
- * and prepend initialization code to each cell execution.
- */
-export class DeepnoteController extends KernelController {
-    constructor(controller: NotebookController) {
-        super(controller);
-    }
-
-    public override createNotebookCellExecution(cell: NotebookCell): NotebookCellExecution {
-        const execution = super.createNotebookCellExecution(cell);
-        return new DeepnoteNotebookCellExecution(execution, cell);
-    }
-}
-
 /**
  * Wrapper around NotebookCellExecution that prepends initialization code.
  * This is implemented by delegating all calls to the underlying execution object.
@@ -89,3 +70,15 @@ class DeepnoteNotebookCellExecution implements NotebookCellExecution {
         return this.execution.appendOutputItems(items, output);
     }
 }
+
+/**
+ * DeepnoteController extends KernelController to intercept cell execution
+ * and prepend initialization code to each cell execution.
+ */
+export class DeepnoteController extends KernelController {
+    public override createNotebookCellExecution(cell: NotebookCell): NotebookCellExecution {
+        const execution = super.createNotebookCellExecution(cell);
+
+        return new DeepnoteNotebookCellExecution(execution, cell);
+    }
+}