-

libs/filer/workspace_files_extensions_client.go

@@ -47,7 +47,12 @@ func (w *workspaceFilesExtensionsClient) getNotebookStatByNameWithExt(ctx contex
 	nameWithoutExt := strings.TrimSuffix(name, string(ext))
 
 	// File name does not have an extension associated with Databricks notebooks, return early.
-	if _, ok := notebook.ExtensionToLanguage[ext]; !ok {
+	if !slices.Contains([]notebook.Extension{
+		notebook.ExtensionPython,
+		notebook.ExtensionR,
+		notebook.ExtensionScala,
+		notebook.ExtensionSql,
+		notebook.ExtensionJupyter}, ext) {
 		return nil, nil
 	}
 

libs/notebook/ext.go

@@ -20,7 +20,6 @@ var ExtensionToLanguage = map[Extension]workspace.Language{
 	ExtensionSql:    workspace.LanguageSql,
 
 	// The platform supports all languages (Python, R, Scala, and SQL) for Jupyter notebooks.
-	ExtensionJupyter: workspace.LanguageUnknown,
 }
 
 func GetExtensionByLanguage(objectInfo *workspace.ObjectInfo) Extension {