"""Contains `DataConverter`."""
+
from typing import List, Union, OrderedDict, Dict, Tuple, Any, Optional, Type
from abc import abstractmethod, ABC
@@ -458,8 +459,7 @@ Source code for graphnet
self._output_files = [
os.path.join(
self._output_dir,
- self._create_file_name(file)
- + self._save_method.file_extension,
+ self._create_file_name(file) + self._save_method.file_extension,
)
for file in input_files
]
@@ -619,9 +619,7 @@ Source code for graphnet
global_index.value += n_ids # type: ignore[name-defined]
else:
starting_index = self._index
- event_nos = np.arange(
- starting_index, starting_index + n_ids, 1
- ).tolist()
+ event_nos = np.arange(starting_index, starting_index + n_ids, 1).tolist()
self._index += n_ids
return event_nos
@@ -681,9 +679,7 @@ Source code for graphnet
[docs]
@final
-
def merge_files(
-
self, files: Optional[List[str]] = None, **kwargs: Any
-
) -> None:
+
def merge_files(self, files: Optional[List[str]] = None, **kwargs: Any) -> None:
"""Merge converted files.
`DataConverter` will call the `.merge_files` method in the
@@ -699,7 +695,7 @@
Source code for graphnet
elif files is not None:
# Proceed to merge specified by user.
if isinstance(files, str):
- files = [files] # Cast to list if user forgot
+ files = [files] # Cast to list if user forgot
files_to_merge = files
else:
# Raise error