style: pre-commit fixes

Author: pre-commit-ci[bot]

jetnet/datasets/dataset.py

@@ -4,7 +4,8 @@
 
 from __future__ import annotations
 
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
 import torch
 from torch import Tensor

jetnet/datasets/jetnet.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
+from collections.abc import Callable
 from copy import copy
-from typing import Callable
 
 import numpy as np
 

jetnet/datasets/qgjets.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
+from collections.abc import Callable
 from copy import copy
-from typing import Callable
 
 import numpy as np
 

jetnet/datasets/toptagging.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
+from collections.abc import Callable
 from copy import copy
-from typing import Callable
 
 import numpy as np