Merge pull request #113 from WilliamDixon/bug/atriumdb-file-loop

Fix: AtriumDB Format is giving an `unable to open database` file error when looping over files
chorus-ai · Nov 14, 2024 · fedbb40 · fedbb40
2 parents d056da0 + 0b8f1b4
commit fedbb40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/waveform_benchmark/formats/atriumdb.py b/waveform_benchmark/formats/atriumdb.py
@@ -20,7 +20,7 @@ class AtriumDB(BaseFormat):
     def write_waveforms(self, path, waveforms):
         # Create a new local dataset using SQLite
         global sdk
-        if sdk is None:
+        if sdk is None or path != sdk.dataset_location:
             sdk = AtriumSDK.create_dataset(dataset_location=path)
             sdk = AtriumSDK(dataset_location=path, num_threads=1)
             sdk.block.block_size = 16384