From 8d42182db831d7a5f321488e35565524bfdf1afb Mon Sep 17 00:00:00 2001
From: Chris LaFreniere <40371649+chlafreniere@users.noreply.github.com>
Date: Thu, 19 Nov 2020 19:54:09 -0800
Subject: [PATCH] Attempt to Colorize Code Cells from Notebook Contents
 (#13473)

* Attempt to colorize from saved language info

* Simplify colorization change

* Fixup
---
 .../services/notebook/browser/models/notebookModel.ts        | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/sql/workbench/services/notebook/browser/models/notebookModel.ts b/src/sql/workbench/services/notebook/browser/models/notebookModel.ts
index fd194c18fd..a7b2371b48 100644
--- a/src/sql/workbench/services/notebook/browser/models/notebookModel.ts
+++ b/src/sql/workbench/services/notebook/browser/models/notebookModel.ts
@@ -336,6 +336,11 @@ export class NotebookModel extends Disposable implements INotebookModel {
 			this._cells = [];
 			if (contents) {
 				this._defaultLanguageInfo = contents.metadata?.language_info;
+				// If language info was serialized in the notebook, attempt to use that to decrease time
+				// required until colorization occurs
+				if (this._defaultLanguageInfo) {
+					this.updateLanguageInfo(this._defaultLanguageInfo);
+				}
 				this._savedKernelInfo = this.getSavedKernelInfo(contents);
 				this._savedConnectionName = this.getSavedConnectionName(contents);
 				if (contents.metadata) {