From 6e4d04450fcefd16ef6273c6ef74f605e0903d56 Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Tue, 6 Aug 2024 16:29:59 +0200 Subject: [PATCH] Update base_graph.py --- scrapegraphai/graphs/base_graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scrapegraphai/graphs/base_graph.py b/scrapegraphai/graphs/base_graph.py index 052d501c..c441f7ab 100644 --- a/scrapegraphai/graphs/base_graph.py +++ b/scrapegraphai/graphs/base_graph.py @@ -157,9 +157,9 @@ class BaseGraph: embedder_model = embedder_model.model if hasattr(current_node, "node_config"): - if type(current_node.node_config) is dict: + if isinstance(current_node.node_config,dict): if current_node.node_config.get("schema", None) and schema is None: - if type(current_node.node_config["schema"]) is not dict: + if not isinstance(current_node.node_config["schema"], dict): # convert to dict try: schema = current_node.node_config["schema"].schema()