[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
CoffeaTeam · Aug 26, 2024 · ec8a56d · ec8a56d
1 parent a692d7b
commit ec8a56d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/coffea/ml_tools/__init__.py b/src/coffea/ml_tools/__init__.py
@@ -6,8 +6,8 @@
 """
 
 from coffea.ml_tools.helper import numpy_call_wrapper
-from coffea.ml_tools.torch_wrapper import torch_wrapper
 from coffea.ml_tools.tf_wrapper import tf_wrapper
+from coffea.ml_tools.torch_wrapper import torch_wrapper
 from coffea.ml_tools.triton_wrapper import triton_wrapper
 from coffea.ml_tools.xgboost_wrapper import xgboost_wrapper
 

diff --git a/src/coffea/ml_tools/helper.py b/src/coffea/ml_tools/helper.py
@@ -159,7 +159,8 @@ class numpy_call_wrapper(abc.ABC):
         {awkward.Array: awkward.to_numpy}, default_conv=container_converter.no_action
     )
     _np_to_ak_ = container_converter(
-        {numpy.ndarray: awkward.from_regular}, default_conv=container_converter.no_action
+        {numpy.ndarray: awkward.from_regular},
+        default_conv=container_converter.no_action,
     )
 
     def __init__(self):