From 45bf2c7da654175ee6e94b87f88bb17b20875674 Mon Sep 17 00:00:00 2001
From: Eugene Brodsky <ebr@users.noreply.github.com>
Date: Thu, 25 Jan 2024 00:17:14 -0500
Subject: [PATCH] chore(updater): address deprecation of pkg_resources

as per module docstring:
This module is deprecated. Users are directed to importlib.resources,
importlib.metadata and packaging instead.
---
 invokeai/frontend/install/invokeai_update.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/invokeai/frontend/install/invokeai_update.py b/invokeai/frontend/install/invokeai_update.py
index c6f06533c5..8c6e8a7afc 100644
--- a/invokeai/frontend/install/invokeai_update.py
+++ b/invokeai/frontend/install/invokeai_update.py
@@ -5,8 +5,8 @@ pip install <path_to_git_source>.
 import os
 import platform
 from distutils.version import LooseVersion
+from importlib.metadata import PackageNotFoundError, distribution
 
-import pkg_resources
 import psutil
 import requests
 from rich import box, print
@@ -89,12 +89,11 @@ def welcome(latest_release: str, latest_prerelease: str):
 
 
 def get_extras():
-    extras = ""
     try:
-        _ = pkg_resources.get_distribution("xformers")
+        distribution("xformers")
         extras = "[xformers]"
-    except pkg_resources.DistributionNotFound:
-        pass
+    except PackageNotFoundError:
+        extras = ""
     return extras