Skip to content

Commit dfa96e8

Browse files
authored
Fix for "Error creating dataset: 'ConnectionPool' object has no attribute 'clone'" in object detection workflow (#121)
1 parent 5628661 commit dfa96e8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

apps/object-detection/app/eval.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,8 +177,9 @@ def main(params):
177177

178178
print(f"Creating dataset...")
179179
try:
180-
dataset = PyTorchDataset.ApertureDBDataset(
181-
pool, q, batch_size=1, label_prop="_uniqueid")
180+
with pool.get_connection() as client:
181+
dataset = PyTorchDataset.ApertureDBDataset(
182+
client, q, batch_size=1, label_prop="_uniqueid")
182183
except Exception as e:
183184
print("Error creating dataset:", e)
184185
dataset = []

0 commit comments

Comments
 (0)