Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 5, 2024
1 parent 3085de9 commit ab4f459
Show file tree
Hide file tree
Showing 5 changed files with 24 additions and 25 deletions.
5 changes: 2 additions & 3 deletions .github/workflows/noise_processing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ on:
workflow_dispatch:
# schedule:
# - cron: '9 * * * *'

jobs:
process:
runs-on: ubuntu-latest
Expand All @@ -22,7 +22,7 @@ jobs:
python -m pip install -U setuptools wheel
python -m pip install -r requirements_noise.txt
- name: Install ffmpeg
run: |
sudo apt-get update
Expand All @@ -43,4 +43,3 @@ jobs:
with:
commit_message: Commit to Github
file_pattern: 'img/*.png'

36 changes: 17 additions & 19 deletions noise_processing.py
Original file line number Diff line number Diff line change
@@ -1,46 +1,44 @@
# importing general Python libraries
import pandas as pd
import datetime as dt
import os

import matplotlib.pyplot as plt
import pytz
import pandas as pd
import plotly.graph_objects as go
import pytz
from orcasound_noise.pipeline.acoustic_util import plot_bb, plot_spec

# importing orcasound_noise libraries
from orcasound_noise.pipeline.pipeline import NoiseAnalysisPipeline
from orcasound_noise.utils import Hydrophone
from orcasound_noise.pipeline.acoustic_util import plot_spec, plot_bb


# Set Location and Resolution
# Port Townsend, 1 Hz Frequency, 60-second samples
if __name__ == '__main__':
pipeline = NoiseAnalysisPipeline(Hydrophone.PORT_TOWNSEND,
delta_f=10, bands=None,
delta_t=60, mode='safe')


if __name__ == "__main__":
pipeline = NoiseAnalysisPipeline(
Hydrophone.PORT_TOWNSEND, delta_f=10, bands=None, delta_t=60, mode="safe"
)


# Generate parquet dataframes with noise levels for a time period

now = dt.datetime.now(pytz.timezone('US/Pacific'))
psd_path, broadband_path = pipeline.generate_parquet_file(now - dt.timedelta(hours = 6),
now - dt.timedelta(hours = 1),
upload_to_s3=False)
now = dt.datetime.now(pytz.timezone("US/Pacific"))
psd_path, broadband_path = pipeline.generate_parquet_file(
now - dt.timedelta(hours=6), now - dt.timedelta(hours=1), upload_to_s3=False
)

# Read the parquet files
psd_df = pd.read_parquet(psd_path)
bb_df = pd.read_parquet(broadband_path)

# Create a new directory if it does not exist
if not os.path.exists('img'):
os.makedirs('img')
if not os.path.exists("img"):
os.makedirs("img")

# Create and save psd plot
# Create and save psd plot
fig = plot_spec(psd_df)
fig.write_image('img/psd.png')
fig.write_image("img/psd.png")

# Create and save bb plot
fig = plot_bb(bb_df)
fig.savefig('img/broadband.png')
fig.savefig("img/broadband.png")
6 changes: 3 additions & 3 deletions requirements_noise.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
matplotlib
scipy
m3u8
matplotlib
scipy
m3u8
git+https://github.com/kkroening/ffmpeg-python
orcasound_noise@git+https://github.com/orcasound/ambient-sound-analysis@plotting_update
kaleido
1 change: 1 addition & 0 deletions tests/test_ooi.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Unit tests for OOI workflow"""

import datetime
import os
import shutil
Expand Down
1 change: 1 addition & 0 deletions tests/test_spectrograms.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Unit tests for various util functions relating to spectrogram creation"""

import os.path

import matplotlib.pyplot as plt
Expand Down

0 comments on commit ab4f459

Please sign in to comment.