Skip to content
This repository was archived by the owner on Dec 20, 2024. It is now read-only.

Commit 7012d5e

Browse files
committed
fix dataframe warning
1 parent ee866b9 commit 7012d5e

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

src/fetch/orderbook.py

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,14 @@ def get_order_rewards(cls, block_range: BlockRange) -> DataFrame:
8888

8989
# Solvers do not appear in both environments!
9090
assert set(prod.solver).isdisjoint(set(barn.solver)), "solver overlap!"
91-
return pd.concat([prod, barn])
91+
if not prod.empty and not barn.empty:
92+
return pd.concat([prod, barn])
93+
elif not prod.empty:
94+
return prod.copy()
95+
elif not barn.empty:
96+
return barn.copy()
97+
else:
98+
return pd.DataFrame()
9299

93100
@classmethod
94101
def get_batch_rewards(cls, block_range: BlockRange) -> DataFrame:
@@ -116,4 +123,11 @@ def get_batch_rewards(cls, block_range: BlockRange) -> DataFrame:
116123

117124
# Solvers do not appear in both environments!
118125
assert set(prod.solver).isdisjoint(set(barn.solver)), "solver overlap!"
119-
return pd.concat([prod, barn])
126+
if not prod.empty and not barn.empty:
127+
return pd.concat([prod, barn])
128+
elif not prod.empty:
129+
return prod.copy()
130+
elif not barn.empty:
131+
return barn.copy()
132+
else:
133+
return pd.DataFrame()

0 commit comments

Comments
 (0)