BUG: imjoy_jupyterlab_extension argument to import_module

Also remove duplicate try: except. Suggested-by: @KumoLiu
thewtex · May 17, 2024 · a352010 · a352010
1 parent 3b172a4
commit a352010
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/itkwidgets/integrations/environment.py b/itkwidgets/integrations/environment.py
@@ -52,15 +52,12 @@ def find_env():
                 # jupyterlab may not be available
                 pass
         try:
-            import_module("imjoy-jupyterlab-extension")
+            import_module("imjoy_jupyterlab_extension")
         except ModuleNotFoundError:
-            try:
-                import_module("imjoy_jupyterlab_extension")
-            except ModuleNotFoundError:
-                if ENVIRONMENT is Env.JUPYTERLITE:
-                    raise RuntimeError('imjoy-jupyterlab-extension is required. Install the package and refresh page.')
-                elif sys.version_info.minor > 7:
-                    raise RuntimeError('imjoy-jupyterlab-extension is required. `pip install itkwidgets[lab]` and refresh page.')
+            if ENVIRONMENT is Env.JUPYTERLITE:
+                raise RuntimeError('imjoy-jupyterlab-extension is required. Install the package and refresh page.')
+            elif sys.version_info.minor > 7:
+                raise RuntimeError('imjoy-jupyterlab-extension is required. `pip install itkwidgets[lab]` and refresh page.')
 
     try:
         import imjoy_elfinder