diff --git a/src/imitation/algorithms/adversarial/common.py b/src/imitation/algorithms/adversarial/common.py index 4dfd554a0..d80005693 100644 --- a/src/imitation/algorithms/adversarial/common.py +++ b/src/imitation/algorithms/adversarial/common.py @@ -2,7 +2,7 @@ import abc import dataclasses import logging -from typing import Iterable, Iterator, Mapping, Optional, Type, List, overload +from typing import Iterable, Iterator, List, Mapping, Optional, Type, overload import numpy as np import torch as th