Fix docgen circular import

logicalclocks · Jul 11, 2024 · ec3ac94 · ec3ac94
1 parent 1cd5cf2
commit ec3ac94
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/python/hsfs/serving_key.py b/python/hsfs/serving_key.py
@@ -16,11 +16,13 @@
 from __future__ import annotations
 
 import json
-from typing import Any, Dict, List, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
 import humps
-from hsfs import feature_group as fg_mod
-from hsfs import util
+
+
+if TYPE_CHECKING:
+    from hsfs import feature_group as fg_mod
 
 
 class ServingKey:
@@ -84,6 +86,8 @@ def to_dict(self) -> Dict[str, Any]:
         }
 
     def __repr__(self) -> str:
+        from hsfs import util
+
         return json.dumps(self, cls=util.FeatureStoreEncoder)
 
     @property