fix embedder loading

kraina-ai · Jun 3, 2023 · 57c0cd7 · 57c0cd7
1 parent f1da952
commit 57c0cd7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/srai/embedders/hex2vec/embedder.py b/srai/embedders/hex2vec/embedder.py
@@ -277,6 +277,8 @@ def load(cls, path: Path) -> "Hex2VecEmbedder":
         Returns:
             Hex2VecEmbedder: The loaded embedder.
         """
+        if isinstance(path, str):
+            path = Path(path)
         with open(path / "config.json") as f:
             embedder_config = json.load(f)
         embedder = cls(**embedder_config["embedder_config"])