Skip to content

Commit

Permalink
Added missing import
Browse files Browse the repository at this point in the history
  • Loading branch information
BloodAxe committed Feb 6, 2025
1 parent 28a3791 commit 07d4100
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_toolbelt/inference/ensembling.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import torch
from torch import nn, Tensor
from typing import List, Union, Iterable, Optional, Dict, Tuple
from typing import List, Union, Iterable, Optional, Dict, Tuple, Mapping

__all__ = [
"ApplySoftmaxTo",
Expand Down

0 comments on commit 07d4100

Please sign in to comment.