fixed regex

Author: DonJayamanne

src/client/jupyter/common/cellHelper.ts

@@ -3,7 +3,7 @@ import { TextDocument, Range } from 'vscode';
 import { JupyterCodeLensProvider } from '../editorIntegration/codeLensProvider';
 import * as vscode from 'vscode';
 
-const CellIdentifier = /^(# %%|#%%|# \<codecell\>|# In\[\d*\]|# In\[ \])(.*)/i;
+const CellIdentifier = /^(# %%|#%%|# \<codecell\>|# In\[\d*?\]|# In\[ \])(.*)/i;
 
 export class CellHelper {
  constructor(private cellCodeLenses: JupyterCodeLensProvider) {

src/client/jupyter/provider.ts

@@ -8,7 +8,7 @@ export class JupyterProvider {
  * @memberOf LanguageProvider
  */
  get cellIdentifier(): RegExp {
- return /^(# %%|#%%|# \<codecell\>|# In\[\d?\]|# In\[ \])(.*)/i;
+ return /^(# %%|#%%|# \<codecell\>|# In\[\d*?\]|# In\[ \])(.*)/i;
  }
 
  /**