Fix python installer launch detection

hrntsm · Aug 7, 2022 · 36cbedc · 36cbedc
1 parent 3c74cdd
commit 36cbedc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Tunny/Util/PythonInstaller.cs b/Tunny/Util/PythonInstaller.cs
@@ -54,8 +54,9 @@ internal static bool CheckPackagesIsInstalled()
             }
             foreach (string package in packageList)
             {
-                string[] singleFilePackages = { "bottle", "optuna-dashboard", "six", "PyYAML", "scikit-learn", "threadpoolctl" };
-                if (!Installer.IsModuleInstalled(package) && !singleFilePackages.Contains(package))
+                string[] singleFilePackages = { "bottle", "optuna-dashboard", "six", "PyYAML", "scikit-learn", "threadpoolctl", "typing_extensions" };
+                string[] useUnderLinePackages = { "opt-einsum", "pyro-api", "pyro-ppl" };
+                if (!Installer.IsModuleInstalled(package) && !singleFilePackages.Contains(package) && !useUnderLinePackages.Contains(package))
                 {
                     return false;
                 }