Skip to content

Commit 76349a9

Browse files
committed
fix empty graph tests
1 parent 7100c05 commit 76349a9

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

Diff for: spikexplore/backends/bluesky.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,9 @@
44
import logging
55
import pandas as pd
66
from datetime import datetime, timedelta, timezone
7+
8+
from atproto_client.exceptions import BadRequestError
9+
710
from spikexplore.NodeInfo import NodeInfo
811
from spikexplore.graph import add_node_attributes, add_edges_attributes
912

@@ -99,8 +102,11 @@ def get_user_skeets(self, username):
99102
user_skeets.items(),
100103
)
101104
return user_skeets, dict(skeets_metadata)
105+
except BadRequestError as e:
106+
logger.error(f"Error in getting user skeets: code {e.response.status_code} - {e.response.content.message}")
107+
return {}, {}
102108
except Exception as e:
103-
logger.error("Error in getting user skeets: ", e)
109+
logger.error(f"Error in getting user skeets: {e}")
104110
return {}, {}
105111

106112
def reshape_node_data(self, node_df):

Diff for: spikexplore/collect_edges.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -134,8 +134,8 @@ def spiky_ball(initial_node_list, graph_handle, cfg, node_acc=NodeInfo(), progre
134134
new_edges = remove_edges_with_target_nodes(new_edges, new_node_list)
135135

136136
new_node_dic, edges_df, nodes_df, node_acc = process_hop(graph_handle, new_node_list, node_acc)
137-
if nodes_df.empty:
138-
break
137+
if edges_df.empty:
138+
continue
139139
nodes_df["spikyball_hop"] = depth # Mark the depth of the spiky ball on the nodes
140140

141141
total_node_list = total_node_list + new_node_list

0 commit comments

Comments
 (0)