From 60fe92263647733c61040882cccf43c9f9c1f750 Mon Sep 17 00:00:00 2001
From: Chigozie Nri <chigozie@stability.ai>
Date: Wed, 28 Sep 2022 12:36:15 +0000
Subject: [PATCH] Test for presence of dotenv before using

---
 src/stability_sdk/client.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/stability_sdk/client.py b/src/stability_sdk/client.py
index 08ca13ee..aa614914 100644
--- a/src/stability_sdk/client.py
+++ b/src/stability_sdk/client.py
@@ -15,11 +15,15 @@
 import grpc
 from argparse import ArgumentParser, Namespace
 from typing import Dict, Generator, List, Optional, Union, Any, Sequence, Tuple
-from dotenv import load_dotenv
 from google.protobuf.json_format import MessageToJson
 from PIL import Image
 
-load_dotenv()
+try:
+    from dotenv import load_dotenv
+except ModuleNotFoundError:
+    pass
+else:
+    load_dotenv()
 
 thisPath = pathlib.Path(__file__).parent.resolve()
 genPath = thisPath / "interfaces/gooseai/generation"