Add installation of preinstall packages for jupyterlite

minjk-bl · minjk-bl · commit 78ae30e07cd0 · 2023-06-30T17:48:20.000+09:00
diff --git a/visualpython/js/com/com_Config.js b/visualpython/js/com/com_Config.js
@@ -296,6 +296,7 @@ define([
          * - automatically restart on jupyter kernel restart (loadVisualpython.js)
          */
         readKernelFunction() {
+            let that = this;
             // CHROME: change method to load py files ($.get -> require)
             return new Promise(function(resolve, reject) {
                 var libraryList = [ 
@@ -314,7 +315,26 @@ define([
                 libraryList.forEach(libCode => {
                     promiseList.push(vpKernel.execute(libCode, true));
                 });
-                
+
+                if (that.extensionType === 'lite') {
+                    let preInstallCode = '';
+                    let preInstallPackList = [
+                        'seaborn',
+                        'plotly',
+                        'scikit-learn',
+                        'scipy',
+                        'statsmodels'
+                    ];
+                    preInstallPackList.forEach((packName, idx) => {
+                        preInstallCode += '%pip install ' + packName
+                        if (idx < preInstallPackList.length - 1) {
+                            preInstallCode += '\n';
+                        }
+                    });
+                    // pre-install packages
+                    promiseList.push(vpKernel.execute(preInstallCode, true));
+                }
+
                 // run all promises
                 let failed = false;
                 Promise.all(promiseList).then(function(resultObj) {
@@ -918,7 +938,7 @@ define([
                                             '- Save VP Note before refreshing the page.'
                                         ];
                                         com_interface.insertCell('markdown', info.join('\n'));
-                                        com_interface.insertCell('code', "import piplite\npiplite.install('jupyterlab-visualpython==" + latestVersion + "')");
+                                        com_interface.insertCell('code', "%pip install jupyterlab-visualpython==" + latestVersion);
                                     }
 
                                     // update version_timestamp
