load plug-ins in try..except block

CHANGELOG.md

@@ -3,6 +3,7 @@
 ### Fixes
 - Fixed operation processor bug if current working directory is different from the script directory ([#1391](https://github.com/neptune-ai/neptune-client/pull/1391))
 - Retry request when ChunkedEncodingError occurred. ([#1402](https://github.com/neptune-ai/neptune-client/pull/1402))
+- Load CLI plug-ins in try..except block to avoid a failure in loading a plug-in to crash entire CLI ([#1392](https://github.com/neptune-ai/neptune-client/pull/1392))
 
 ### Features
 - Added support for `tensorboard` integration ([#1368](https://github.com/neptune-ai/neptune-client/pull/1368))

src/neptune/cli/__main__.py

@@ -14,6 +14,8 @@
 # limitations under the License.
 #
 
+import warnings
+
 import click
 import pkg_resources
 
@@ -36,4 +38,11 @@ def main():
 plugins = {entry_point.name: entry_point for entry_point in pkg_resources.iter_entry_points("neptune.plugins")}
 
 for name, entry_point in plugins.items():
-    main.add_command(entry_point.load(), name)
+    # loading an entry_point may fail and this
+    # will cause all CLI commands to fail.
+    # So, we load the plug-ins in try and except block.
+    try:
+        loaded_plugin = entry_point.load()
+    except Exception as e:
+        warnings.warn(f"Failed loading neptune plug-in `{name}` with exception: {e}")
+    main.add_command(loaded_plugin, name)