From 4495d90b37c4a87176869de2993ed43c109304ae Mon Sep 17 00:00:00 2001
From: Omar2535 <omar@9tsai.ca>
Date: Sun, 17 Nov 2024 20:20:35 -0500
Subject: [PATCH] Update imports

---
 graphqler/core.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/graphqler/core.py b/graphqler/core.py
index 29e491e..d1af396 100644
--- a/graphqler/core.py
+++ b/graphqler/core.py
@@ -3,7 +3,7 @@
 from graphqler.utils.config_handler import set_config
 from graphqler.utils.stats import Stats
 from graphqler.utils.objects_bucket import ObjectsBucket
-
+from graphqler.utils.api import API
 
 def compile_and_fuzz(path: str, url: str, input_config: dict | None = None) -> dict:
     """
@@ -40,7 +40,9 @@ def compile_and_fuzz(path: str, url: str, input_config: dict | None = None) -> d
     fuzzer = Fuzzer(path, url)
     fuzzer.run()
 
-    objects_bucket = ObjectsBucket(api=fuzzer.api).load()
+    api: API = fuzzer.api
+
+    objects_bucket = ObjectsBucket(api=api).load()
     stats = Stats().load()
 
     return {