Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 0 additions & 18 deletions benchmarks/benchmarks/generate_data/um_files.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,31 +24,13 @@ def _create_um_files(
from datetime import datetime
from tempfile import NamedTemporaryFile

from mo_pack import compress_wgdos as mo_pack_compress
from mule import ArrayDataProvider, Field3, FieldsFile
from mule.pp import fields_to_pp_file
import numpy as np

from iris import load_cube
from iris import save as save_cube

def packing_patch(*compress_args, **compress_kwargs) -> bytes:
"""
Force conversion from returned :class:`memoryview` to :class:`bytes`.

Downstream uses of :func:`mo_pack.compress_wgdos` were written
for the ``Python2`` behaviour, where the returned buffer had a
different ``__len__`` value to the current :class:`memoryview`.
Unable to fix directly in Mule, so monkey patching for now.
"""
return mo_pack_compress(*compress_args, **compress_kwargs).tobytes()

import mo_pack

mo_pack.compress_wgdos = packing_patch

########

template = {
"fixed_length_header": {"dataset_type": 3, "grid_staggering": 3},
"integer_constants": {
Expand Down