Skip to content

Commit 05444ac

Browse files
committed
correct unit MiB
1 parent b248121 commit 05444ac

5 files changed

Lines changed: 7 additions & 7 deletions

File tree

src/birdnet/acoustic/inference/core/encoding/encoding_result.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,9 @@ def memory_size_MiB(self) -> float:
7474
"""Return the total result memory usage including embeddings buffers.
7575
7676
Returns:
77-
float: Memory size in megabytes.
77+
float: Memory size in mebibytes.
7878
"""
79-
return super().memory_size_mb + (
79+
return super().memory_size_MiB + (
8080
(
8181
self._embeddings.nbytes
8282
+ self._embeddings_masked.nbytes

src/birdnet/acoustic/inference/core/result_base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -179,9 +179,9 @@ def memory_size_MiB(self) -> float:
179179
"""Memory usage for the base result metadata.
180180
181181
Returns:
182-
float: Memory used by metadata buffers in megabytes.
182+
float: Memory used by metadata buffers in mebibytes.
183183
"""
184-
return super().memory_size_mb + (
184+
return super().memory_size_MiB + (
185185
(
186186
self._inputs.nbytes
187187
+ self._input_durations.nbytes

src/birdnet_tests/acoustic_models/inference/encoding/encoding_result_py/test_encoding_io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def test_save_and_load_is_equal() -> None:
9595
)
9696

9797

98-
def test_memory_size_mb() -> None:
98+
def test_memory_size_MiB() -> None:
9999
res = create_dummy_result()
100100

101101
expected_size = (

src/birdnet_tests/acoustic_models/inference/predictions/prediction_result_py/test_io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def test_save_and_load_is_equal() -> None:
4949
)
5050

5151

52-
def test_memory_size_mb() -> None:
52+
def test_memory_size_MiB() -> None:
5353
res = create_file_prediction_result(
5454
n_files=3,
5555
duration_s=54,

src/birdnet_tests/geo_models/inference/scores_result_py/test_geo_io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def get_dummy_result() -> GeoPredictionResult:
2323
return res
2424

2525

26-
def test_memory_size_mb() -> None:
26+
def test_memory_size_MiB() -> None:
2727
res = get_dummy_result()
2828

2929
expected_size = (

0 commit comments

Comments
 (0)