update to guard in case package not installed

triton-inference-server · Jan 6, 2024 · 4c1ed33 · 4c1ed33
1 parent 565bf9d
commit 4c1ed33
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/python/library/tritonclient/__init__.py b/src/python/library/tritonclient/__init__.py
@@ -26,8 +26,13 @@
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-from importlib.metadata import version
+from importlib.metadata import PackageNotFoundError, version
 
-__version__ = version("tritonclient")
+__version__ = "UNKNOWN"
+
+try:
+    __version__ = version("tritonclient")
+except PackageNotFoundError:
+    pass
 
 del version