Skip to content

Commit e0d547e

Browse files
committed
drop output data instead of trying to remove it.
Drop handles cases such as if the data does not exist etc.
1 parent 8dfaccc commit e0d547e

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
@@ -120,7 +120,7 @@ def output_datasets(self) -> list[Dataset]:
120120
def drop_artifacts(self):
121121
for out_data in self.out_data:
122122
if out_data is not None:
123-
rmtree(out_data.store)
123+
out_data.drop()
124124

125125
def init(self):
126126
self.init_out_array()

0 commit comments

Comments
 (0)