Fix sizing cell incorrectly for webview outputs

- webview outputs are not "hasDynamicHeight"
- The "else if" condition got flipped around in the last change

src/vs/workbench/contrib/notebook/browser/view/renderers/codeCell.ts

@@ -325,7 +325,7 @@ export class CodeCell extends Disposable {
 				if (renderedOutput.renderResult.type !== RenderOutputType.None) {
 					// Show inset in webview, or render output that isn't rendered
 					// TODO@roblou skipHeightInit flag is a hack - the webview only sends the real height once. Don't wipe it out here.
-					this.renderOutput(currOutput, index, undefined, true);
+					this.renderOutput(currOutput, index, undefined);
 				} else {
 					// Anything else, just update the height
 					this.viewCell.updateOutputHeight(index, renderedOutput.element.clientHeight);
@@ -430,7 +430,7 @@ export class CodeCell extends Disposable {
 		);
 	}
 
-	private renderOutput(currOutput: IProcessedOutput, index: number, beforeElement?: HTMLElement, skipHeightInit = false) {
+	private renderOutput(currOutput: IProcessedOutput, index: number, beforeElement?: HTMLElement) {
 		if (!this.outputResizeListeners.has(currOutput)) {
 			this.outputResizeListeners.set(currOutput, new DisposableStore());
 		}
@@ -510,7 +510,6 @@ export class CodeCell extends Disposable {
 			outputItemDiv.style.position = 'absolute';
 		}
 
-		if (!skipHeightInit) {
 		if (outputHasDynamicHeight(result)) {
 			this.viewCell.selfSizeMonitoring = true;
 
@@ -539,7 +538,7 @@ export class CodeCell extends Disposable {
 			elementSizeObserver.startObserving();
 			this.outputResizeListeners.get(currOutput)!.add(elementSizeObserver);
 			this.viewCell.updateOutputHeight(index, clientHeight);
-			} else if (result.type !== RenderOutputType.None) { // no-op if it's a webview
+		} else if (result.type === RenderOutputType.None) { // no-op if it's a webview
 			const clientHeight = Math.ceil(outputItemDiv.clientHeight);
 			this.viewCell.updateOutputHeight(index, clientHeight);
 
@@ -547,7 +546,6 @@ export class CodeCell extends Disposable {
 			outputItemDiv.style.top = `${top}px`;
 		}
 	}
-	}
 
 	generateRendererInfo(renderId: string | undefined): string {
 		if (renderId === undefined || renderId === BUILTIN_RENDERER_ID) {

src/vs/workbench/contrib/notebook/common/notebookCommon.ts

@@ -313,7 +313,7 @@ export interface IRenderOutputViaExtension {
 export type IInsetRenderOutput = IRenderPlainHtmlOutput | IRenderOutputViaExtension;
 export type IRenderOutput = IRenderNoOutput | IInsetRenderOutput;
 
-export const outputHasDynamicHeight = (o: IRenderOutput) => o.type === RenderOutputType.Extension || o.hasDynamicHeight;
+export const outputHasDynamicHeight = (o: IRenderOutput) => o.type !== RenderOutputType.Extension && o.hasDynamicHeight;
 
 export type NotebookCellTextModelSplice = [
 	number /* start */,