Skip to content

Commit

Permalink
housekeeping
Browse files Browse the repository at this point in the history
- typo fix
- add comment for where segment weights maybe added
  • Loading branch information
kkappler committed Jan 17, 2024
1 parent e8e44e5 commit 542d377
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
4 changes: 3 additions & 1 deletion aurora/pipelines/process_mth5.py
Original file line number Diff line number Diff line change
Expand Up @@ -148,6 +148,7 @@ def process_tf_decimation_level(
frequency_bands = config.decimations[i_dec_level].frequency_bands_obj()
transfer_function_obj = TTFZ(i_dec_level, frequency_bands, processing_config=config)
dec_level_config = config.decimations[i_dec_level]
# segment_weights = coherence_weights(dec_level_config, local_stft_obj, remote_stft_obj)
transfer_function_obj = process_transfer_functions(
dec_level_config, local_stft_obj, remote_stft_obj, transfer_function_obj
)
Expand Down Expand Up @@ -315,6 +316,7 @@ def process_mth5(
tfk.make_processing_summary()
tfk.show_processing_summary()
tfk.validate()

# See Note #1
if tfk.config.decimations[0].save_fcs:
mth5_mode = "a"
Expand Down Expand Up @@ -404,7 +406,7 @@ def process_mth5(
if z_file_path:
tf_cls.write(z_file_path)

tfk.dataset.close_mths_objs()
tfk.dataset.close_mth5s()
if return_collection:
# this is now really only to be used for debugging and may be deprecated soon
return tf_collection
Expand Down
5 changes: 2 additions & 3 deletions aurora/transfer_function/kernel_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -449,10 +449,9 @@ def get_run_object(self, index_or_row):
run_obj = row.mth5_obj.from_reference(row.run_reference)
return run_obj

def close_mths_objs(self):
def close_mth5s(self):
"""
Loop over all unique mth5_objs in the df and make sure they are closed
Loop over all unique mth5_objs in dataset df and make sure they are closed.+
"""
mth5_objs = self.df["mth5_obj"].unique()
for mth5_obj in mth5_objs:
Expand Down

0 comments on commit 542d377

Please sign in to comment.