vscode-jupyter: KQL syntax stopped working

Environment data

• Language Server version: 2021.6.2-pre.1
• OS and version: linux x64
• Python version (and distribution if applicable, e.g. Anaconda):
• python.analysis.indexing: undefined
• python.analysis.typeCheckingMode: off

Expected behaviour

Can use %kql

Actual behaviour

Pylance puts errors everywhere.

Code Snippet / Additional information

%%kql RetainedByCohort <<
let cohortWindow = 7d;