Skip to content

Commit 723fcfa

Browse files
committed
call drop method on out_data
this should handle cases where out_data doesn't exist
1 parent ea60080 commit 723fcfa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/volara_torch/blockwise/predict.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def output_datasets(self) -> list[Dataset]:
112112
def drop_artifacts(self):
113113
for out_data in self.out_data:
114114
if out_data is not None:
115-
rmtree(out_data.store)
115+
out_data.drop()
116116

117117
def init(self):
118118
self.init_out_array()

0 commit comments

Comments
 (0)