From c43d9178ca1bc653a3da4b6a207cc7559625544a Mon Sep 17 00:00:00 2001 From: Robert Jackson Date: Fri, 13 Jan 2023 14:05:29 -0700 Subject: [PATCH] FIX: pre-commit pass --- xbatcher/generators.py | 13 +++++++------ xbatcher/tests/test_generators.py | 5 +++-- 2 files changed, 10 insertions(+), 8 deletions(-) diff --git a/xbatcher/generators.py b/xbatcher/generators.py index 2d4fb25..8252992 100644 --- a/xbatcher/generators.py +++ b/xbatcher/generators.py @@ -1,9 +1,8 @@ """Classes for iterating through xarray datarrays / datasets in batches.""" import itertools -import warnings import json - +import warnings from operator import itemgetter from typing import Any, Dict, Hashable, Iterator, List, Optional, Sequence, Union @@ -285,9 +284,11 @@ def to_json(self): out_member_dict = {} member_keys = [x for x in member.keys()] for member_key in member_keys: - out_member_dict[member_key] = {'start': member[member_key].start, - 'stop': member[member_key].stop, - 'step': member[member_key].step} + out_member_dict[member_key] = { + "start": member[member_key].start, + "stop": member[member_key].stop, + "step": member[member_key].step, + } out_dict["selector"] = out_member_dict return json.dumps(out_dict) @@ -301,7 +302,7 @@ def to_file(self, out_file_name: str): The path to the json file to write to. """ out_json = self.to_json() - out_file = open(out_file_name, mode='w') + out_file = open(out_file_name, mode="w") out_file.write(out_json) out_file.close() diff --git a/xbatcher/tests/test_generators.py b/xbatcher/tests/test_generators.py index 708aead..25d7db2 100644 --- a/xbatcher/tests/test_generators.py +++ b/xbatcher/tests/test_generators.py @@ -1,8 +1,9 @@ +import json +import tempfile + import numpy as np import pytest import xarray as xr -import tempfile -import json from xbatcher import BatchGenerator, BatchSchema from xbatcher.testing import (